fffiloni commited on
Commit
586a940
·
1 Parent(s): 043d9fd

Update app.py

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