Fixes to UI

app.py

@@ -27,6 +27,7 @@ model = Predictor(n_labels=len(LABELS))
 
 model_file = hf_hub_download(
     repo_id="Hali5/Mae-Model-MedMNIST-Predictor",
+        # Create a read-only text box to display the column for Truth Labels
     filename="checkpoints/model_linear_v2_epoch_100.pt"
 )
 
@@ -88,7 +89,6 @@ with gradio.Blocks() as demo:
     with gradio.Tab("Examples"):
         gradio.Markdown("Click an example below to test the model against the PathMNIST test dataset.")
         
-        # Create a read-only text box to display the column for Truth Labels
         with gradio.Column():
                 input_img_ex = gradio.Image(label="Selected Test Image")
                 truth_box = gradio.Textbox(label="Ground Truth Label", interactive=False)
@@ -97,8 +97,8 @@ with gradio.Blocks() as demo:
             output_lbl_ex = gradio.Label(num_top_classes=3, label="Model Prediction")
         
         gradio.Examples(
-            examples=example_rows,  # Passes both the image file path and the text label
-            inputs=[input_img_ex, truth_box],  # Maps the data columns to both UI components
+            examples=example_rows, 
+            inputs=[input_img_ex, truth_box],
             outputs=output_lbl_ex,
             fn=predict,
             cache_examples=True,