added v7 & v8 comparision

app.py

@@ -173,7 +173,10 @@ def inference2(video,model_link,iou_threshold,confidence_threshold):
     finalVideo.release()
     return 'output.mp4',np.mean(fps_video)
 
-
+def inference_comp(image,iou_threshold,confidence_threshold):
+    v8_out, v8_fps = inference(image, "yolov8m",image_comp_iou_threshold,image_comp_conf_threshold)
+    v7_out, v7_fps = inference(image, "yolov7",image_comp_iou_threshold,image_comp_conf_threshold)
+    return v7_out,v8_out,v7_fps,v8_fps
 
 examples_images = ['data/images/1.jpg',
             'data/images/2.jpg',
@@ -208,9 +211,22 @@ with gr.Blocks() as demo:
         gr.Examples(examples=examples_videos,inputs=video_input,outputs=video_output)
         video_button = gr.Button("Detect")
     
-    # with gr.Tab("Webcam Video"):
-    #     gr.Markdown("## YOLOv7 Inference on Webcam Video")
-    #     gr.Markdown("Coming Soon")
+    with gr.Tab("Compare Models"):
+        gr.Markdown("## YOLOv7 vs YOLOv8 Object detection comparision")
+        with gr.Row():
+            image_comp_input = gr.Image(type='pil', label="Input Image", source="upload")
+        text_comp_button = gr.Button("Detect")
+        gr.Examples(examples=examples_images,inputs=image_input,outputs=image_output)
+        with gr.Row():
+            image_comp_iou_threshold = gr.Slider(label="IOU Threshold",interactive=True, minimum=0.0, maximum=1.0, value=0.45)
+            image_comp_conf_threshold = gr.Slider(label="Confidence Threshold",interactive=True, minimum=0.0, maximum=1.0, value=0.25)
+        with gr.Row():
+            image_comp_output_v7 = gr.Image(type='pil', label="YOLOv7 Output Image", source="upload")
+            image_comp_output_v8 = gr.Image(type='pil', label="YOLOv8 Output Image", source="upload")
+        with gr.row():
+            v7_fps_image = gr.Number(0,label='v7 FPS')        
+            v8_fps_image = gr.Number(0,label='v8 FPS')
+        
 
     text_button.click(inference, inputs=[image_input,image_drop,
                                          image_iou_threshold,image_conf_threshold],
@@ -218,5 +234,9 @@ with gr.Blocks() as demo:
     video_button.click(inference2, inputs=[video_input,video_drop,
                                            video_iou_threshold,video_conf_threshold],            
                                         outputs=[video_output,fps_video])
+    text_comp_button.click(inference_comp,input=[image_comp_input,
+                                            image_comp_iou_threshold,
+                                            image_comp_conf_threshold],
+                                            outputs=[image_comp_output_v7,image_comp_output_v8,v7_fps_image,v8_fps_image])
 
 demo.launch(debug=True,enable_queue=True)