mskov commited on
Commit
f57fca3
Β·
1 Parent(s): c6ad7d5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Aud
23
 
24
  print(dataset, "and at 0[audio][array] ", dataset[0]["audio"]["array"], type(dataset[0]["audio"]["array"]), "and at audio : ", dataset[0]["audio"])
25
 
26
- model = WhisperForConditionalGeneration.from_pretrained("mskov/miso_test").to("cuda")
27
 
28
  # Evaluate the model
29
  # model.eval()
 
23
 
24
  print(dataset, "and at 0[audio][array] ", dataset[0]["audio"]["array"], type(dataset[0]["audio"]["array"]), "and at audio : ", dataset[0]["audio"])
25
 
26
+ model = WhisperForConditionalGeneration.from_pretrained("mskov/whisper-small-esc50").to("cuda")
27
 
28
  # Evaluate the model
29
  # model.eval()