Update app.py
Browse files
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(
|
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
|