Parthebhan commited on
Commit
1556f7a
·
verified ·
1 Parent(s): 5370777

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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-e.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-e.pt", local_dir="./")
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