fffiloni commited on
Commit
f7f4ced
1 Parent(s): 84d7e81

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -195,7 +195,7 @@ def infer():
195
  write_jpeg(flow_img, f"predicted_flow.jpg")
196
  flo_file = write_flo(predicted_flow, "flofile.flo")
197
 
198
- warp_res = warp(img1_batc, img2_batc, predicted_flow, blend=0.5)
199
  return "done", "predicted_flow.jpg", ["flofile.flo"], warp_res
200
  ####################################
201
  # Bonus: Creating GIFs of predicted flows
 
195
  write_jpeg(flow_img, f"predicted_flow.jpg")
196
  flo_file = write_flo(predicted_flow, "flofile.flo")
197
 
198
+ warp_res = warp(img1_batch, img2_batch, predicted_flow, blend=0.5)
199
  return "done", "predicted_flow.jpg", ["flofile.flo"], warp_res
200
  ####################################
201
  # Bonus: Creating GIFs of predicted flows