Update app.py

app.py

@@ -3,13 +3,12 @@ from huggingface_hub import hf_hub_download
 import os
 import cv2  # Import OpenCV
 
-# Function to download models from Hugging Face
 def download_models(model_id):
     # Check if the model file exists locally
-    local_model_path = "./ gelan-c-seg.pt"
+    local_model_path = "./gelan-c-seg.pt"
     if not os.path.exists(local_model_path):
         # Download the model from Hugging Face if it doesn't exist locally
-        hf_hub_download("merve/yolov9", filename=" gelan-c-seg.pt", local_dir="./")
+        hf_hub_download(model_id, filename="gelan-c-seg.pt", local_dir="./")
     return local_model_path
 
 
@@ -20,7 +19,7 @@ def yolov9_inference(img):
     :return: Output image with detections.
     """
     # Load the model
-    model_path = download_models("gelan-c-seg.pt")
+    model_path = download_models("merve/yolov9")
     # Assuming you're using a YOLOv9 model from Ultralytics, you would typically use their library to load the model
     model = cv2.dnn.readNetFromDarknet(model_path)
     # Perform inference
@@ -34,7 +33,7 @@ def yolov9_inference(img):
 def app():
     return gr.Interface(
         fn=yolov9_inference,
-        inputs= gr.Image(type="filepath", label="Image"),
+        inputs=gr.inputs.Image(type="file", label="Image"),
         outputs="image",
         title="YOLOv9 Inference",
         description="Perform object detection using the YOLOv9 model.",