Update app.py

app.py

@@ -227,7 +227,9 @@ def infer():
     # res = cv2.remap(img, flow, None, cv2.INTER_LINEAR)
     res = cv2.remap(frame1pil, numpy_array_flow, None, cv2.INTER_LANCZOS4)
     print(res)
-    return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg'
+    res = Image.fromarray(res)
+    res.save('wraped.jpg')
+    return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg', 'wraped.jpg'
 ####################################
 # Bonus: Creating GIFs of predicted flows
 # ---------------------------------------
@@ -256,4 +258,4 @@ def infer():
 # ffmpeg -f image2 -framerate 30 -i predicted_flow_%d.jpg -loop -1 flow.gif
 
       
-gr.Interface(fn=infer, inputs=[], outputs=[gr.Textbox(), gr.Image(), gr.Files(), gr.Image()]).launch()
+gr.Interface(fn=infer, inputs=[], outputs=[gr.Textbox(), gr.Image(), gr.Files(), gr.Image(), gr.Image()]).launch()