mskov commited on
Commit
be77284
Β·
1 Parent(s): 74ce8f3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -18,12 +18,13 @@ disable_caching()
18
  huggingface_token = os.environ["huggingface_token"]
19
  pipe = pipeline(model="mskov/whisper-small-esc50")
20
  print(pipe)
21
- processor = WhisperProcessor.from_pretrained("openai/whisper-medium").to("cuda")
22
  dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio(sampling_rate=16000))
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 = pipe
 
27
  # Evaluate the model
28
  # model.eval()
29
  #print("model.eval ", model.eval())
 
18
  huggingface_token = os.environ["huggingface_token"]
19
  pipe = pipeline(model="mskov/whisper-small-esc50")
20
  print(pipe)
21
+ processor = WhisperProcessor.from_pretrained("mskov/whisper-small-esc50")
22
  dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Audio(sampling_rate=16000))
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()
30
  #print("model.eval ", model.eval())