update

app.py

@@ -6,7 +6,11 @@ import torch
 print(torch.__version__)
 print(torch.version.cuda)
 
-os.system('export DISPLAY=:0.0; Xvfb :0 -screen 0 640x480x24 &')
+os.system('mkdir -p /home/user/app/tmp/.X11-unix')
+os.system('chmod 1777 /home/user/app/tmp/.X11-unix')
+os.environ["TMPDIR"] = "/home/user/app/tmp"
+os.environ["DISPLAY"] = ":0.0"
+os.system('Xvfb :0 -screen 0 640x480x24 &')
 
 # download model
 print("Downloading model weights")