Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -6,10 +6,10 @@ import cv2 # Import OpenCV
|
|
6 |
# Function to download models from Hugging Face
|
7 |
def download_models(model_id):
|
8 |
# Check if the model file exists locally
|
9 |
-
local_model_path = "./gelan-
|
10 |
if not os.path.exists(local_model_path):
|
11 |
# Download the model from Hugging Face if it doesn't exist locally
|
12 |
-
hf_hub_download("merve/yolov9", filename="gelan-
|
13 |
return local_model_path
|
14 |
|
15 |
|
|
|
6 |
# Function to download models from Hugging Face
|
7 |
def download_models(model_id):
|
8 |
# Check if the model file exists locally
|
9 |
+
local_model_path = "./ gelan-c-seg.pt"
|
10 |
if not os.path.exists(local_model_path):
|
11 |
# Download the model from Hugging Face if it doesn't exist locally
|
12 |
+
hf_hub_download("merve/yolov9", filename=" gelan-c-seg.pt", local_dir="./")
|
13 |
return local_model_path
|
14 |
|
15 |
|