Update app.py
Browse files
app.py
CHANGED
@@ -400,7 +400,7 @@ elif feature == "LA - Image Segmetation":
|
|
400 |
# βββ PREDICTION ββββββββββββββββββββββββββββββββββββ
|
401 |
if img0 is not None and run_la:
|
402 |
img_np = np.array(img0)
|
403 |
-
model = YOLO('
|
404 |
with st.spinner("Running YOLO modelβ¦"):
|
405 |
results = model(img_np, imgsz=640)
|
406 |
|
|
|
400 |
# βββ PREDICTION ββββββββββββββββββββββββββββββββββββ
|
401 |
if img0 is not None and run_la:
|
402 |
img_np = np.array(img0)
|
403 |
+
model = YOLO('best_100.pt') # path to your weights
|
404 |
with st.spinner("Running YOLO modelβ¦"):
|
405 |
results = model(img_np, imgsz=640)
|
406 |
|