Update app.py

app.py

@@ -24,7 +24,7 @@ model = YOLO('foduucom/stockmarket-future-prediction')
 #############################################################Image Inference############################################################
 def yolov8_img_inference(
     image: gr.inputs.Image = None,
-    #model_path: gr.inputs.Dropdown = None,
+    model_path: gr.inputs.Dropdown = None,
     image_size: gr.inputs.Slider = 640,
     conf_threshold: gr.inputs.Slider = 0.25,
     iou_threshold: gr.inputs.Slider = 0.45,
@@ -40,7 +40,7 @@ def yolov8_img_inference(
     Returns:
         Rendered image
     """
-    #model = YOLO(model_path)
+    model = YOLO(model_path)
     model.overrides['conf'] = conf_threshold
     model.overrides['iou']= iou_threshold
     model.overrides['agnostic_nms'] = False  # NMS class-agnostic