Spaces:

IbrahimaThioye
/

sam_yolo_flask

Running

IbrahimaThioye commited on Nov 19, 2024

Commit

51a0d38

verified ·

1 Parent(s): 0ffee70

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -59,7 +59,7 @@ except Exception as e:
 try:
     sam2_checkpoint = app.config['SAM_2']
-    model_cfg = "configs/sam2.1/sam2.1_hiera_l.yaml"
     sam2_model = build_sam2(model_cfg, sam2_checkpoint, device="cpu")
     predictor = SAM2ImagePredictor(sam2_model)

 try:
     sam2_checkpoint = app.config['SAM_2']
+    model_cfg = "configs/sam2.1/sam2.1_hiera_t.yaml"
     sam2_model = build_sam2(model_cfg, sam2_checkpoint, device="cpu")
     predictor = SAM2ImagePredictor(sam2_model)