Update app.py

app.py

@@ -10,8 +10,6 @@ from yolov8 import xai_yolov8s
 import time
 import tempfile
 
-
-
 # Sample images directory
 sample_images = {
     "Sample 1": os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
@@ -46,7 +44,6 @@ def process_image(sample_choice, uploaded_image, yolo_versions):
 
     return result_images
 
-
 def view_model(selected_models):
     for model in selected_models[0]:
         if model == "yolov5":
@@ -60,10 +57,7 @@ def view_model(selected_models):
             </iframe>
             """
             return iframe_html
-        return "<p>Error: Unable to generate Netron visualization.</p>"
-    return "<p>Please upload a valid model file.</p>"
-
-
+    return "<p>Please select a valid model for Netron visualization.</p>"
 
 # Custom CSS for styling (optional)
 custom_css = """
@@ -91,10 +85,10 @@ with gr.Blocks(css=custom_css) as interface:
                 value=default_sample,
             )
 
-            #upload_image = gr.Image(
-            #    label="Upload an Image",
-            #    type="pil",  
-           # )
+            upload_image = gr.Image(
+                label="Upload an Image",
+                type="pil",  
+            )
 
             selected_models = gr.CheckboxGroup(
                 choices=["yolov5", "yolov8s"],
@@ -119,7 +113,6 @@ with gr.Blocks(css=custom_css) as interface:
             height=500,
         )
 
-        #model_file_input = gr.File(label="Upload Model File", type="filepath")  # Fixed type here
         netron_display = gr.HTML(label="Netron Visualization")
 
     sample_selection.change(
@@ -134,7 +127,7 @@ with gr.Blocks(css=custom_css) as interface:
         outputs=[result_gallery],
     )
 
-    model_file_input.change(
+    selected_models.change(
         fn=view_model,
         inputs=selected_models,
         outputs=netron_display,