Update app.py

app.py

@@ -23,7 +23,7 @@ dataset = load_dataset("mskov/miso_test", split="test").cast_column("audio", Aud
 
 print(dataset, "and at 0[audio][array] ", dataset[0]["audio"]["array"], type(dataset[0]["audio"]["array"]), "and at audio : ", dataset[0]["audio"])
 
-model = WhisperForConditionalGeneration.from_pretrained("mskov/whisper-small-esc50").to("cuda")
+model = WhisperForConditionalGeneration.from_pretrained("mskov/whisper-small-esc50")
 
 # Evaluate the model
 # model.eval()
@@ -35,7 +35,7 @@ def map_to_pred(batch):
     
     
     with torch.no_grad():
-        predicted_ids = model.generate(input_features.to("cuda"))[0]
+        predicted_ids = model.generate(input_features)[0]
         
     transcription = processor.decode(predicted_ids)
     batch["prediction"] = processor.tokenizer._normalize(transcription)