Update app.py

app.py

@@ -1,11 +1,17 @@
 import gradio as gr
 import spaces
 from huggingface_hub import hf_hub_download
+import os
 
-
+# Function to download models from Hugging Face
 def download_models(model_id):
-    hf_hub_download("merve/yolov9", filename=f"{model_id}", local_dir=f"./")
-    return f"./{model_id}"
+    # Check if the model file exists locally
+    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="./")
+    return local_model_path
+
 
 @spaces.GPU
 def yolov9_inference(img_path, model_id, image_size, conf_threshold, iou_threshold):