Gradio 4.44.1 compatibility: simplify Dataframe params and bump README sdk_version

gradio_two_stage_app.py

@@ -82,14 +82,10 @@ def predict(image: Image.Image):
 
     # Build probability tables as 2D lists
     bin_probs = binary_part.get('probs', []) or []
-    bin_rows = _as_table_rows(getattr(get_cascade(), 'bin_id2label', {}), bin_probs)
-    bin_rows = _ensure_rows(bin_rows)
+    bin_rows = _ensure_rows(_as_table_rows(getattr(get_cascade(), 'bin_id2label', {}), bin_probs))
 
-    multi_rows = []
-    if 'multiclass' in result:
-        m_probs = multi_part.get('probs', []) or []
-        multi_rows = _as_table_rows(getattr(get_cascade(), 'multi_id2label', {}), m_probs)
-    multi_rows = _ensure_rows(multi_rows)
+    m_probs = multi_part.get('probs', []) or []
+    multi_rows = _ensure_rows(_as_table_rows(getattr(get_cascade(), 'multi_id2label', {}), m_probs))
 
     return "\n".join(summary_lines), bin_rows, multi_rows
 
@@ -114,23 +110,13 @@ def build_interface():
         with gr.Tab("Binary Probabilities"):
             bin_table = gr.Dataframe(
                 headers=["label", "probability"],
-                datatype=["str", "number"],
-                row_count=3,
-                col_count=(2, "fixed"),
-                type="array",
                 value=[["", 0.0], ["", 0.0], ["", 0.0]],
-                interactive=False,
                 label="Binary Stage Probabilities",
             )
         with gr.Tab("Origin Probabilities"):
             multi_table = gr.Dataframe(
                 headers=["label", "probability"],
-                datatype=["str", "number"],
-                row_count=3,
-                col_count=(2, "fixed"),
-                type="array",
                 value=[["", 0.0], ["", 0.0], ["", 0.0]],
-                interactive=False,
                 label="Origin Stage Probabilities",
             )