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