Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -160,7 +160,7 @@ def detect(img,model):
|
|
| 160 |
print(f'{s}Done. ({t2 - t1:.3f}s)')
|
| 161 |
show_seg_result(im0, (da_seg_mask,ll_seg_mask), is_demo=True)
|
| 162 |
|
| 163 |
-
|
| 164 |
weights = 'End-to-end.pth'
|
| 165 |
print(weights)
|
| 166 |
|
|
|
|
| 160 |
print(f'{s}Done. ({t2 - t1:.3f}s)')
|
| 161 |
show_seg_result(im0, (da_seg_mask,ll_seg_mask), is_demo=True)
|
| 162 |
|
| 163 |
+
if weights == 'yolop.pt':
|
| 164 |
weights = 'End-to-end.pth'
|
| 165 |
print(weights)
|
| 166 |
|