hank1996 commited on
Commit
5d628fa
·
1 Parent(s): 2f2e7f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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
- elif weights == 'yolop.pt':
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