bmay commited on
Commit
47c37ba
·
verified ·
1 Parent(s): d850561

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -43,8 +43,9 @@ def run_theia(image):
43
  vis_video = np.stack(
44
  [np.vstack([tr, gtr]) for tr, gtr in zip(theia_decode_results, gt_decode_results, strict=False)]
45
  )
 
46
 
47
  return vis_video
48
 
49
- demo = gr.Interface(fn=run_theia, inputs=gr.Image(type="pil"), outputs=gr.Image(type="pil"))
50
  demo.launch()
 
43
  vis_video = np.stack(
44
  [np.vstack([tr, gtr]) for tr, gtr in zip(theia_decode_results, gt_decode_results, strict=False)]
45
  )
46
+ vis_video = (255.0 * vis_video).astype(np.uint8)
47
 
48
  return vis_video
49
 
50
+ demo = gr.Interface(fn=run_theia, inputs=gr.Image(type="pil"), outputs=gr.Image(type="numpy"))
51
  demo.launch()