Update app.py

app.py

@@ -19,18 +19,18 @@ Intrigued? Dive into the world of trading alchemy! 🌌
 image_path= [['test/1.jpg', 'foduucom/stockmarket-future-prediction', 640, 0.25, 0.45], ['test/2.jpg', 'foduucom/stockmarket-future-prediction', 640, 0.25, 0.45],['test/3.jpg', 'foduucom/stockmarket-future-prediction', 640, 0.25, 0.45]]
 
 # Load YOLO model
-#model = YOLO("foduucom/stockmarket-future-prediction")
+model = YOLO("foduucom/stockmarket-future-prediction")
 
 
 #############################################################Image Inference############################################################
 def yolov8_img_inference(
     image: gr.inputs.Image = None,
-    model_path: gr.inputs.Dropdown = ["foduucom/stockmarket-future-prediction"],
+    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
 ):
-    model = YOLO(model_path[0])
+    model = YOLO(model_path)
     model.overrides['conf'] = conf_threshold
     model.overrides['iou']= iou_threshold
     model.overrides['agnostic_nms'] = False  # NMS class-agnostic