Spaces:

Woziii
/

scribe

Running

Woziii commited on Aug 18, 2024

Commit

4bf7cc5

verified ·

1 Parent(s): 0ba180b

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ MODEL_NAME = "openai/whisper-medium"
 FILE_LIMIT_MB = 1000
 YT_LENGTH_LIMIT_S = 3600
-device = 0 if torch.cuda.is_available() else "cpu"
 pipe = pipeline(
     task="automatic-speech-recognition",

 FILE_LIMIT_MB = 1000
 YT_LENGTH_LIMIT_S = 3600
+device = "cuda:0" if torch.cuda.is_available() else "cpu"
 pipe = pipeline(
     task="automatic-speech-recognition",