Fix sam link
Browse files
app.py
CHANGED
@@ -253,7 +253,7 @@ def build_model_lora(pipe_concept, lora_paths, style_path, condition, args, pipe
|
|
253 |
def build_yolo_segment_model(sam_path, device):
|
254 |
yolo_world = YOLOWorld(model_id="yolo_world/l")
|
255 |
sam = EfficientViTSamPredictor(
|
256 |
-
create_sam_model(name=
|
257 |
)
|
258 |
return yolo_world, sam
|
259 |
|
|
|
253 |
def build_yolo_segment_model(sam_path, device):
|
254 |
yolo_world = YOLOWorld(model_id="yolo_world/l")
|
255 |
sam = EfficientViTSamPredictor(
|
256 |
+
create_sam_model(name='efficientvit_sam_xl1.pt', weight_url=sam_path).to(device).eval()
|
257 |
)
|
258 |
return yolo_world, sam
|
259 |
|