bmay commited on
Commit
4342a51
·
verified ·
1 Parent(s): f4fd721

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -4
app.py CHANGED
@@ -40,10 +40,8 @@ def run_theia(image):
40
  device='cuda',
41
  )
42
 
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
 
 
40
  device='cuda',
41
  )
42
 
43
+ vis = np.vstack([theia_decode_results[0], gt_decode_results[0]])
44
+ vis = (255.0 * vis).astype(np.uint8)
 
 
45
 
46
  return vis_video
47