Update app.py
Browse files
app.py
CHANGED
@@ -206,8 +206,8 @@ def infer():
|
|
206 |
# display the PIL image
|
207 |
#img.show()
|
208 |
img.save('frame_input.jpg')
|
209 |
-
|
210 |
-
|
211 |
return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
|
212 |
####################################
|
213 |
# Bonus: Creating GIFs of predicted flows
|
|
|
206 |
# display the PIL image
|
207 |
#img.show()
|
208 |
img.save('frame_input.jpg')
|
209 |
+
res = get_warp_res("frame_input.jpg", "flofile.flo", fname_output='warped.png')
|
210 |
+
print(res)
|
211 |
return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
|
212 |
####################################
|
213 |
# Bonus: Creating GIFs of predicted flows
|