updated process_image args

app.py

@@ -21,7 +21,7 @@ def load_sample_image(sample_name):
         return Image.open(image_path)
     return None
 
-def process_image(sample_choice, uploaded_image, yolo_versions):
+def process_image(sample_choice, uploaded_image, yolo_versions, target_lyr = -5, n_components = 8):
     """Process the image using selected YOLO models."""
     # Load sample or uploaded image
     if uploaded_image is not None:
@@ -37,7 +37,7 @@ def process_image(sample_choice, uploaded_image, yolo_versions):
     # Apply selected models
     for yolo_version in yolo_versions:
         if yolo_version == "yolov5":
-            result_images.append(xai_yolov5(image)) 
+            result_images.append(xai_yolov5(image, target_lyr = -5, n_components = 8)) 
         elif yolo_version == "yolov8s":
             result_images.append(xai_yolov8s(image))
         else:
@@ -129,14 +129,14 @@ with gr.Blocks(css=custom_css) as interface:
 
 
     # Multi-threaded processing
-    def run_both(sample_choice, uploaded_image, selected_models):
+    def run_both(sample_choice, uploaded_image, selected_models, target_lyr = -5, n_components = 8):
         results = []
         netron_html = ""
 
         # Thread to process the image
         def process_thread():
             nonlocal results
-            results = process_image(sample_choice, uploaded_image, selected_models)
+            results = process_image(sample_choice, uploaded_image, selected_models, target_lyr = -5, n_components = 8)
 
         # Thread to generate Netron visualization
         def netron_thread():
@@ -158,7 +158,7 @@ with gr.Blocks(css=custom_css) as interface:
     # Run button click
     run_button.click(
         fn=run_both,
-        inputs=[sample_selection, upload_image, selected_models],
+        inputs=[sample_selection, upload_image, selected_models, target_lyr = -5, n_components = 8],
         outputs=[result_gallery, netron_display, dff_gallery],
     )