Update app.py

app.py

@@ -8,9 +8,8 @@ torch.set_default_device("cuda")
 
 # # Ensure GPU usage if available
 # device = "cuda" if torch.cuda.is_available() else "cpu"
-# torch.set_default_tensor_type('torch.cuda.FloatTensor' if device=='cuda' else 'torch.FloatTensor')
 
-torch.set_default_tensor_type('torch.cuda.FloatTensor')
+torch.set_default_dtype()('torch.cuda.FloatTensor')