Simplify Gradio schema and disable API docs

app.py

@@ -63,21 +63,22 @@ def run_demo(image: Image.Image, candidate_text: str):
         scores = (image_outputs.image_embeds @ text_outputs.text_embeds.t()).squeeze(0)
         probs = scores.softmax(dim=-1)
 
-    table = [
-        [label, float(score), float(prob)]
+    rows = [
+        (label, float(score), float(prob))
         for label, score, prob in zip(labels, scores.tolist(), probs.tolist())
     ]
-    table.sort(key=lambda row: row[2], reverse=True)
+    rows.sort(key=lambda row: row[2], reverse=True)
 
-    top_label = table[0][0]
-    top_prob = table[0][2]
+    top_label = rows[0][0]
+    top_prob = rows[0][2]
     summary = f"Top match: {top_label} ({top_prob:.4f})"
-    raw = {
-        "labels": labels,
-        "scores": scores.tolist(),
-        "probs": probs.tolist(),
+    details = {
+        "ranked_results": [
+            {"label": label, "score": score, "prob": prob}
+            for label, score, prob in rows
+        ]
     }
-    return summary, table, json.dumps(raw, ensure_ascii=False, indent=2)
+    return summary, json.dumps(details, ensure_ascii=False, indent=2)
 
 
 def build_demo():
@@ -103,17 +104,14 @@ def build_demo():
 
         run_button = gr.Button("Run Matching", variant="primary")
         summary_output = gr.Textbox(label="Summary")
-        table_output = gr.Dataframe(
-            headers=["label", "score", "prob"],
-            datatype=["str", "number", "number"],
-            label="Results",
-        )
-        json_output = gr.Code(label="Raw Output", language="json")
+        details_output = gr.Textbox(label="Results JSON", lines=18)
 
         run_button.click(
             run_demo,
             inputs=[image_input, labels_input],
-            outputs=[summary_output, table_output, json_output],
+            outputs=[summary_output, details_output],
+            api_name=False,
+            show_api=False,
         )
     return demo
 
@@ -126,4 +124,4 @@ except ModuleNotFoundError:
 if __name__ == "__main__":
     if demo is None:
         raise RuntimeError("gradio is required to launch this app.")
-    demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
+    demo.launch(server_name="0.0.0.0", server_port=7860, show_api=False)