enable zerogpu

app.py

@@ -37,10 +37,10 @@ def set_seed(seed):
     torch.cuda.manual_seed_all(seed)
     random.seed(seed)
 
-if torch.cuda.is_available():
-    device = "cuda"
-else:
-    device = "cpu"
+# if torch.cuda.is_available():
+device = "cuda"
+# else:
+    # device = "cpu"
 
 def remove_prefix(text, prefix):
     if text.startswith(prefix):