try cpu only

app.py

@@ -31,7 +31,7 @@ from rocketshp.network import (
 )
 
 os.environ["OMP_NUM_THREADS"] = "4"
-os.environ["CUDA_VISIBLE_DEVICES"] = "0"
+os.environ["CUDA_VISIBLE_DEVICES"] = ""
 
 
 def plot_predictions(
@@ -138,7 +138,8 @@ 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 = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")
+    device = "cpu"
     model = RocketSHP.load_from_checkpoint(model_variant).to(device)
     is_sequence_model = "seq" in model_variant or "mini" in model_variant