add cuda

app.py

@@ -5,11 +5,11 @@ from transformers import pipeline
 @spaces.GPU(duration=60)
 
 # Check if GPU is available; fallback to CPU if not
-device = 0 if torch.cuda.is_available() else -1
+# device = 0 if torch.cuda.is_available() else -1
 
 try:
     # Load models with error handling
-    transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=device)
+    transcriber = pipeline("automatic-speech-recognition", model="openai/whisper-base", device=cuda)
     summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 except Exception as e:
     print(f"Error loading models: {e}")