IbrahimaThioye commited on
Commit
8a61aa8
·
verified ·
1 Parent(s): 51a0d38

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -33,8 +33,7 @@ class Config:
33
  AREA_DATA_FOLDER = os.path.join(BASE_DIR, 'static', 'yolo','area_data')
34
  ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}
35
  MAX_CONTENT_LENGTH = 16 * 1024 * 1024 # 16MB max file size
36
- SAM_CHECKPOINT = os.path.join(BASE_DIR, 'static', 'sam',"sam_vit_h_4b8939.pth")
37
- SAM_2 = os.path.join(BASE_DIR, 'static', 'sam',"sam2.1_hiera_large.pt")
38
  YOLO_PATH = os.path.join(BASE_DIR, 'static', 'yolo', "model_yolo.pt")
39
  RETRAINED_MODEL_PATH = os.path.join(BASE_DIR, 'static', 'yolo', "model_retrained.pt")
40
  DATA_PATH = os.path.join(BASE_DIR, 'static', 'yolo','dataset_yolo', "data.yaml")
 
33
  AREA_DATA_FOLDER = os.path.join(BASE_DIR, 'static', 'yolo','area_data')
34
  ALLOWED_EXTENSIONS = {'png', 'jpg', 'jpeg'}
35
  MAX_CONTENT_LENGTH = 16 * 1024 * 1024 # 16MB max file size
36
+ SAM_2 = os.path.join(BASE_DIR, 'static', 'sam',"sam2.1_hiera_tiny.pt")
 
37
  YOLO_PATH = os.path.join(BASE_DIR, 'static', 'yolo', "model_yolo.pt")
38
  RETRAINED_MODEL_PATH = os.path.join(BASE_DIR, 'static', 'yolo', "model_retrained.pt")
39
  DATA_PATH = os.path.join(BASE_DIR, 'static', 'yolo','dataset_yolo', "data.yaml")