Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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="
|
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()
|