Spaces:
Runtime error
Runtime error
Commit
·
754b1cc
1
Parent(s):
034b367
Update app.py
Browse files
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
|
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 |
|