Update app.py

app.py

@@ -69,7 +69,9 @@ def preprocess_and_predict(sketch_data, rotation_val, noise_val, blur_val):
 
     class_labels = [str(i) for i in range(10)] + ["Unknown"]
     distribution = {class_labels[i]: float(probabilities[i]) for i in range(11)}
-    preview_output = np.clip(np_img, 0, 255).astype(np.uint8)
+    
+    preview_pil = Image.fromarray(np.clip(np_img, 0, 255).astype(np.uint8))
+    preview_output = preview_pil.resize((280, 280), Image.Resampling.NEAREST)
 
     return preview_output, distribution
 
@@ -89,7 +91,7 @@ with gr.Blocks(title="Robust MNIST Classifier") as interface:
             run_btn = gr.Button("Evaluate Signature", variant="primary")
             
         with gr.Column():
-            preview = gr.Image(label="Model-View Reconstruction (28x28)", image_mode="L")
+            preview = gr.Image(label="Model-View Reconstruction (280x280)", image_mode="L")
             probabilities_output = gr.Label(num_top_classes=5, label="Probability Map Output")
 
     run_btn.click(