Update app.py

app.py

@@ -74,7 +74,7 @@ def transcribe_30_seconds(audio_file):
             predicted_ids = model.generate(
                 input_features,
                 attention_mask=attention_mask,
-                max_new_tokens=444,   # keep under total token limit
+                max_new_tokens=444,
                 do_sample=False,
                 forced_decoder_ids=forced_decoder_ids
             )
@@ -82,21 +82,15 @@ def transcribe_30_seconds(audio_file):
         text_chunk = processor.batch_decode(predicted_ids, skip_special_tokens=True)[0]
         partial_text += text_chunk + "\n"
 
-        # Send updated partial text to the UI
         yield partial_text
 
-# Build Gradio UI
-with gr.Blocks() as demo:
-    gr.Markdown("## Hebrew Whisper (Truncate to 30s, No Progress Bar)")
-
-    audio_input = gr.Audio(type="filepath", label="Upload Audio (Truncate to 30s)")
-    output_text = gr.Textbox(label="Partial Transcription")
-
-    start_btn = gr.Button("Start Transcription")
-    stop_btn = gr.Button("Stop Processing", variant="stop")
-
-    # Stream chunk-by-chunk, no progress bar
-    start_btn.click(transcribe_30_seconds, inputs=audio_input, outputs=output_text)
-    stop_btn.click(stop)
+# Build Gradio UI with API support
+demo = gr.Interface(
+    fn=transcribe_30_seconds,
+    inputs=gr.Audio(type="filepath"),
+    outputs="text",
+    title="Hebrew Whisper API",
+    api_name="transcribe"  # Enables API access
+)
 
 demo.launch()