Update app.py

app.py

@@ -16,15 +16,14 @@ forced_decoder_ids = processor.get_decoder_prompt_ids(language="italian", task="
 # Custom preprocessing function
 def preprocess_audio(audio_data):
     # Apply any custom preprocessing to the audio data here if needed
-    return processor(audio_data, return_tensors="pt").input_features
+    # Ensure that the input data is a valid format for the model
+    processed_data = processor(audio_data, return_tensors="pt", padding=True, truncation=True)
+    return processed_data
 
 # Function to perform ASR on audio data
-def transcribe_audio(audio_data):
-    # Preprocess the audio data
-    input_features = preprocess_audio(audio_data)
-
+def transcribe_audio(input_features):
     # Generate token ids
-    predicted_ids = model.generate(input_features)
+    predicted_ids = model.generate(**input_features)
     
     # Decode token ids to text
     transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)