Update app.py

app.py

@@ -34,20 +34,24 @@ def transcribe_audio(audio_url):
         inputs = processor(chunk, sampling_rate=16000, return_tensors="pt", padding=True)
         input_features = inputs.input_features.to(device)
 
-        predicted_ids = model.generate(input_features)
+        with torch.no_grad():
+            predicted_ids = model.generate(input_features, forced_decoder_ids=forced_decoder_ids)
+
         transcription = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
         partial_text += transcription + "\n"
 
-    return partial_text
+    return partial_text.strip()
 
 @app.route('/transcribe', methods=['POST'])
 def transcribe_endpoint():
     data = request.get_json()
-    audio_url = data['audio_url']
+    audio_url = data.get('audio_url')
+    if not audio_url:
+        return jsonify({"error": "Missing 'audio_url' in request"}), 400
 
     transcription = transcribe_audio(audio_url)
 
-    return {"transcription": transcription}
+    return jsonify({"transcription": transcription})
 
 if __name__ == '__main__':
-    app.run(host="0.0.0.0", port=8080)
+    app.run(host="0.0.0.0", port=7860)