Spaces:

CAMB-AI
/

mars5_space

Runtime error

arnavmehta7 commited on Jun 17, 2024

Commit

26fdaed

verified ·

1 Parent(s): 3a0629c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ asr_model = pipeline(
     "automatic-speech-recognition",
     model="openai/whisper-medium",
     chunk_length_s=30,
-    device=torch.device("cuda"),
 )
 def transcribe_file(f: str) -> str:

     "automatic-speech-recognition",
     model="openai/whisper-medium",
     chunk_length_s=30,
+    device=torch.device("cuda:0"),
 )
 def transcribe_file(f: str) -> str: