Update app.py
Browse files
app.py
CHANGED
@@ -173,8 +173,8 @@ def infer():
|
|
173 |
# output_folder = "/tmp/" # Update this to the folder of your choice
|
174 |
write_jpeg(flow_img, f"predicted_flow.jpg")
|
175 |
flo_file = write_flo(predicted_flow, "flofile.flo")
|
176 |
-
|
177 |
-
warp_res = warp(frames[100], frames[101], "flofile.flo",
|
178 |
return "done", "predicted_flow.jpg", ["flofile.flo"], warp_res
|
179 |
####################################
|
180 |
# Bonus: Creating GIFs of predicted flows
|
|
|
173 |
# output_folder = "/tmp/" # Update this to the folder of your choice
|
174 |
write_jpeg(flow_img, f"predicted_flow.jpg")
|
175 |
flo_file = write_flo(predicted_flow, "flofile.flo")
|
176 |
+
|
177 |
+
warp_res = warp(frames[100], frames[101], "flofile.flo", blend=0.5)
|
178 |
return "done", "predicted_flow.jpg", ["flofile.flo"], warp_res
|
179 |
####################################
|
180 |
# Bonus: Creating GIFs of predicted flows
|