Update app.py

app.py

@@ -32,13 +32,18 @@ subprocess.run(shlex.split('pip install wheel/torchmcubes-0.1.0-cp310-cp310-linu
 from tsr.system import TSR
 from tsr.utils import remove_background, resize_foreground, to_gradio_3d_orientation
 
+os.environ["CUDA_VISIBLE_DEVICES"] = "1"
 
 HEADER = """FRAME AI"""
 
-if torch.cuda.is_available():
-    device = "cuda:1"
-else:
-    device = "cpu"
+# if torch.cuda.is_available():
+#     device = "cuda:0"
+# else:
+#     device = "cpu"
+
+# torch.cuda.set_device(1)
+
+CUDA_LAUNCH_BLOCKING=1
 
 model = TSR.from_pretrained(
     "stabilityai/TripoSR",