Upload app.py

2.CNN/app.py

@@ -725,7 +725,7 @@ def enrich_diagnostics(stats, probs):
     return stats
 
 
-def predict_number(main_canvas, stroke_scale):
+def predict_number(main_canvas):
     ensure_model_loaded()
     composed, seg_diag = compose_from_single_canvas(main_canvas)
     if composed is None:
@@ -737,7 +737,6 @@ def predict_number(main_canvas, stroke_scale):
 
     result = preprocess_composed_28x56(
         composed,
-        stroke_scale=stroke_scale,
         extra_diag=seg_diag,
     )
     if result is None:
@@ -775,15 +774,8 @@ with gr.Blocks() as demo:
     )
 
     with gr.Row():
-        with gr.Column(scale=1):
+        with gr.Column(scale=1) as left_col:
             main_canvas = gr.Sketchpad(label="Draw Two Digits (00–99)")
-            stroke_slider = gr.Slider(
-                minimum=0.3,
-                maximum=1.2,
-                value=1.0,
-                step=0.05,
-                label="Stroke Intensity (scale)",
-            )
 
         with gr.Column(scale=1):
             pred_box = gr.Number(label="Predicted Number", precision=0, value=None)
@@ -796,29 +788,32 @@ with gr.Blocks() as demo:
             preview = gr.Image(label="Model Input Preview (28x56)", image_mode="L")
             mean_diff_view = gr.Image(label="Difference vs Training Mean", image_mode="L")
             diagnostics_box = gr.Code(label="Diagnostics (JSON)", language="json")
-            predict_btn = gr.Button("Predict", variant="primary")
-            clear_btn = gr.ClearButton(
-                [
-                    main_canvas,
-                    stroke_slider,
-                    pred_box,
-                    prob_table,
-                    preview,
-                    mean_diff_view,
-                    diagnostics_box,
-                ]
-            )
+
+        # Place buttons under the canvas, but wire them to clear outputs as well
+        with left_col:
+            with gr.Row():
+                predict_btn = gr.Button("Predict", variant="primary")
+                clear_btn = gr.ClearButton(
+                    [
+                        main_canvas,
+                        pred_box,
+                        prob_table,
+                        preview,
+                        mean_diff_view,
+                        diagnostics_box,
+                    ]
+                )
 
     predict_btn.click(
         fn=predict_number,
-        inputs=[main_canvas, stroke_slider],
+        inputs=[main_canvas],
         outputs=[pred_box, prob_table, preview, mean_diff_view, diagnostics_box],
     )
     # On Spaces, avoid per-stroke inference to prevent event floods
     if not IS_SPACE:
         main_canvas.change(
             fn=predict_number,
-            inputs=[main_canvas, stroke_slider],
+            inputs=[main_canvas],
             outputs=[pred_box, prob_table, preview, mean_diff_view, diagnostics_box],
         )