Update app.py

app.py

@@ -12,12 +12,14 @@ import sf3d.utils as sf3d_utils
 from gradio_litmodel3d import LitModel3D
 from huggingface_hub import login
 import subprocess
-device = "cuda" if torch.cuda.is_available() else "cpu"
+
 dtype = torch.bfloat16
 
 torch.backends.cuda.matmul.allow_tf32 = True
 huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
 
+device = torch.device('cuda')
+
 import shutil
 
 def find_cuda():