Future-Tense commited on
Commit
754b1cc
·
1 Parent(s): 034b367

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -70,12 +70,14 @@ def pl_fn(cap,cur,last,fps,pl_tog):
70
  player.set(cv2.CAP_PROP_POS_FRAMES, cur)
71
  ret, frame_bgr = player.read(cur)
72
  frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)
 
 
73
  if pl_tog ==1:
74
  #ins_cnt+=1
75
  cur+=1
76
  else:
77
  cur = cur
78
- return frame,cur
79
 
80
 
81
 
 
70
  player.set(cv2.CAP_PROP_POS_FRAMES, cur)
71
  ret, frame_bgr = player.read(cur)
72
  frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)
73
+ results = model.predict(frame)
74
+ render = render_result(model=model, image=frame, result=results[0])
75
  if pl_tog ==1:
76
  #ins_cnt+=1
77
  cur+=1
78
  else:
79
  cur = cur
80
+ return render,cur
81
 
82
 
83