Update app.py

app.py

@@ -1,18 +1,35 @@
+print("import torch")
 import torch
+
+print("import gradio")
 import gradio as gr
+
+print("import huggingface_hub")
 from huggingface_hub import hf_hub_download
+
+print("import PIL")
 from PIL import Image, ImageDraw
+
+print("import numpy")
 import numpy as np
+
+print("import json")
 import json
+
+print("import opencv")
 import cv2
+
+print("import scipy")
 from scipy.ndimage import gaussian_filter
 
+
 # Constants and Model Downloads
 REPO_ID = "thoucentric/Shelf_Objects_Detection_Yolov7_Pytorch"
 FILENAME = "best.pt"
 yolov7_custom_weights = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
 
 # Load YOLOv7 Custom Model
+print("Load YOLOv7 Custom Model")
 model = torch.hub.load('Owaiskhan9654/yolov7-1:main', model='custom', path_or_model=yolov7_custom_weights, force_reload=True)
 
 # Image Splitting and Merging Functionality
@@ -196,36 +213,42 @@ inputs = [
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.25, step=0.01, label="Confidence Threshold"),
     gr.inputs.Slider(minimum=0.0, maximum=1.0, default=0.45, step=0.01, label="IOU Threshold"),
 ]
-outputs_image = gr.outputs.Image(type="pil", label="Output Image")
+outputs_image = [
+    gr.outputs.Image(type="pil", label="Output Image")
+]
 outputs_json = gr.Textbox(label="Bounding Boxes JSON")
 
 title = "<center>Yolov7 Custom Object Detection</center>"
 description = "<center>Nolen Felten</center>"
 footer = ("<br><br><center><b>Item Classes it will detect (Total 140 Classes)</b></center>")
 
-# Regular Object Detection Interface
-interface = gr.Interface(
-    fn=object_detection,
-    inputs=inputs,
-    outputs=[outputs_image, outputs_json],
-    title=title,
-    description=description,
-    article=footer,
-    cache_examples=False,
-    allow_flagging="never"
-)
 
-# Edge Enhanced Object Detection Interface
-interface_edge = gr.Interface(
-    fn=object_detection_with_edge_enhancement,
-    inputs=inputs,
-    outputs=[outputs_image, outputs_json],
-    title="Object Detection with Edge Enhancement",
-    description=description,
-    article=footer,
-    cache_examples=False,
-    allow_flagging="never"
-)
+interfaces = [
+    # Regular Object Detection Interface
+    gr.Interface(
+        fn=object_detection,
+        inputs=inputs,
+        outputs=[outputs_image, outputs_json],
+        title=title,
+        description=description,
+        article=footer,
+        cache_examples=False,
+        allow_flagging="never"
+    ), 
+    
+
+    # Edge Enhanced Object Detection Interface
+    gr.Interface(
+        fn=object_detection_with_edge_enhancement,
+        inputs=inputs,
+        outputs=[outputs_image, outputs_json],
+        title="Object Detection with Edge Enhancement",
+        description=description,
+        article=footer,
+        cache_examples=False,
+        allow_flagging="never"
+    )
+]
 
 # Edge Enhanced Density-Based Counting Interface
 inputs_density_edge = [