re-enable gpu support

app.py

@@ -31,7 +31,7 @@ from rocketshp.network import (
 )
 
 os.environ["OMP_NUM_THREADS"] = "4"
-os.environ["CUDA_VISIBLE_DEVICES"] = ""
+os.environ["CUDA_VISIBLE_DEVICES"] = "0"
 
 
 def plot_predictions(
@@ -138,8 +138,7 @@ def predict_rocketshp(
         raise gr.Error("Failed to authorize repository access.")
 
     # Load the model
-    # device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
-    device = "cpu"
+    device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
     model = RocketSHP.load_from_checkpoint(model_variant).to(device)
     is_sequence_model = "seq" in model_variant or "mini" in model_variant