Spaces:

mskov
/

test

Runtime error

mskov commited on Sep 5, 2023

Commit

6b7e790

1 Parent(s): 7843939

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -16,7 +16,7 @@ from transformers import WhisperForConditionalGeneration, WhisperProcessor
 processor = WhisperProcessor.from_pretrained("mskov/whisper-small-esc50")
-model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-small-esc50").to("cuda")
 def map_to_pred(batch):
     audio = batch["audio"]

 processor = WhisperProcessor.from_pretrained("mskov/whisper-small-esc50")
+model = WhisperForConditionalGeneration.from_pretrained("mskov/whisper-small-esc50").to("cuda")
 def map_to_pred(batch):
     audio = batch["audio"]