Update app.py

app.py

@@ -3,7 +3,7 @@ import spaces
 from huggingface_hub import hf_hub_download
 
 def download_models(model_id):
-    hf_hub_download("merve/yolov9", filename=f"{model_id}", local_dir=f"./")
+    hf_hub_download("qbcode/chicken_detection", filename=f"{model_id}", local_dir=f"./")
     return f"./{model_id}"
 
 @spaces.GPU
@@ -20,11 +20,12 @@ def yolov9_inference(img_path, model_id, image_size, conf_threshold, iou_thresho
     :return: Annotated image with the number of objects detected.
     """
     # Import YOLOv9
-    import yolov9
+    import yolov8
+    
     
     # Load the model
     model_path = download_models(model_id)
-    model = yolov9.load(model_path, device="cuda:0")
+    model = yolov8.load(model_path, device="cuda:0")
     
     # Set model parameters
     model.conf = conf_threshold
@@ -83,8 +84,8 @@ def app():
                 output_image = gr.Image(type="numpy",label="Output")
                 num_objects_detected = gr.Textbox(label="Number of Objects Detected", readonly=True)
 
-        yolov9_infer.click(
-            fn=yolov9_inference,
+        yolov8_infer.click(
+            fn=yolov8_inference,
             inputs=[
                 img_path,
                 model_path,
@@ -112,7 +113,7 @@ def app():
                     0.5,
                 ],
             ],
-            fn=yolov9_inference,
+            fn=yolov8_inference,
             inputs=[
                 img_path,
                 model_path,