Update app.py

app.py

@@ -17,14 +17,14 @@ model_mms = VitsModel.from_pretrained("facebook/mms-tts-nld")
 tokenizer_mms = VitsTokenizer.from_pretrained("facebook/mms-tts-nld")
 
 processor = WhisperProcessor.from_pretrained("openai/whisper-base")
-model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base").to(device)
+model = WhisperForConditionalGeneration.from_pretrained("openai/whisper-base")
 forced_decoder_ids = processor.get_decoder_prompt_ids(language="nl", task="transcribe")
 sampling_rate = processor.feature_extractor.sampling_rate
 
 
 def translate(audio):
     input_features = processor(audio,sampling_rate=sampling_rate,return_tensors="pt").input_features
-    predicted_ids = model.generate(input_features.to(device),forced_decoder_ids=forced_decoder_ids)
+    predicted_ids = model.generate(input_features,forced_decoder_ids=forced_decoder_ids)
     transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)
     return transcription