Update app.py

app.py

@@ -25,8 +25,7 @@ def preprocess_audio(audio_data, sampling_rate=16_000):
 def transcribe_audio(audio_data):
     input_features = preprocess_audio(audio_data)
     input_values = torch.tensor(input_features["input_values"]).unsqueeze(0)  # Add batch dimension
-    # Ensure the input tensor has the correct shape
-    input_values = input_values.view(1, 1, -1)
+    input_values = input_values.view(1, -1)  # Flatten the tensor to 2D
     predicted_ids = model.generate(input_values)
     transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)
     return transcription[0]