Future-Tense commited on
Commit
aaed971
·
1 Parent(s): 00f3fbc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -45,7 +45,7 @@ def pl_fn(cap,cur,last,fps,pl_tog):
45
  player.set(cv2.CAP_PROP_POS_FRAMES, cur)
46
  ret, frame_bgr = player.read(cur)
47
  #frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)
48
- frame = Image.open(frame_bgr)
49
  output = segmentor(frame)
50
  for i in range(len(output)):
51
  mask = np.array(output[i]['mask'])/255
@@ -73,7 +73,7 @@ def pl_fn(cap,cur,last,fps,pl_tog):
73
  return results,cur
74
 
75
  with gr.Blocks() as app:
76
- gr.Markdown("""<center><h1>Slow Video Object Detection</h1><h4>Gradio and ultralyticsplus/yolov8s</h4><h4>Probably faster on GPU 🤷‍♂️</h4></center>""")
77
  play_state = gr.Markdown("""<right><h7></h7></right>""")
78
 
79
  with gr.Row():
 
45
  player.set(cv2.CAP_PROP_POS_FRAMES, cur)
46
  ret, frame_bgr = player.read(cur)
47
  #frame = cv2.cvtColor(frame_bgr, cv2.COLOR_BGR2RGB)
48
+ frame = Image.fromarray(frame_bgr)
49
  output = segmentor(frame)
50
  for i in range(len(output)):
51
  mask = np.array(output[i]['mask'])/255
 
73
  return results,cur
74
 
75
  with gr.Blocks() as app:
76
+ gr.Markdown("""<center><h1>Testing</h1><h4>stuff</h4></center>""")
77
  play_state = gr.Markdown("""<right><h7></h7></right>""")
78
 
79
  with gr.Row():