Update app.py

app.py

@@ -1,5 +1,8 @@
-import os
-os.system('export CUDA_VISIBLE_DEVICES=""')
+import torch
+if torch.cuda.is_available():
+    torch.cuda.set_device(torch.device('cpu'))
+else:
+    torch.device('cpu')
 
 import subprocess
 import sys