Update app.py

app.py

@@ -45,13 +45,14 @@ with gr.Blocks(title="Code Summarization") as demo:
         desc = gr.Textbox(label="Description", placeholder="What does the code do?")
     code = gr.Textbox(lines=8, label="Code", placeholder="Paste your code here...")
 
-    btn = gr.Button("Generate Summaries (5 beams)")
+    btn = gr.Button("Generate Summaries")
     out_table = gr.Dataframe(headers=["#", "Summary"], label="Generated Summaries", interactive=False)
 
     btn.click(
         summarize_one,
         inputs=[lang, desc, code],
-        outputs=[out_table]
+        outputs=[out_table],
+        api_name="predict"
     )
 
     gr.Markdown(f"**Model:** `{REPO_ID}`  •  **Input max length:** {MAX_LENGTH}  •  **Output max length:** {GEN_MAX_LENGTH}  •  **num_beams:** 5")