Update app.py

app.py

@@ -5,7 +5,7 @@ import torch
 import uuid
 
 # Set device
-# device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda" if torch.cuda.is_available() else "cpu"
 # dtype = torch.float16 if device == "cuda" else torch.float32
 dtype = torch.float32