Update app.py

app.py

@@ -6,11 +6,11 @@ import numpy as np
 import gradio as gr
 
 def _grab_best_device(use_gpu=True):
-    #if torch.cuda.device_count() > 0 and use_gpu:
-    #    device = "cuda"
-    #else:
-    #    device = "cpu"
-    device = 0 if torch.cuda.is_available() else -1
+    if torch.cuda.device_count() > 0 and use_gpu:
+        device = 0 #"cuda"
+    else:
+        device = -1 #"cpu"
+    #device = 0 if torch.cuda.is_available() else -1
     
     return device