Update app.py

app.py

@@ -198,7 +198,7 @@ def infer():
     transform = T.ToPILImage()
     
     # convert the tensor to PIL image using above transform
-    img = transform(frames[100])
+    img = transform(frames[101])
     img = img.resize((960, 520))
     # display the PIL image
     #img.show()
@@ -230,7 +230,9 @@ def infer():
     print(res)
     res = Image.fromarray(res)
     res.save('wraped.jpg')
-    return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg', 'wraped.jpg'
+    image4 = Image.blend(res,frame1pil,0.5)
+    image4.save("output2.jpg")
+    return "done", "predicted_flow.jpg", ["flofile.flo"], 'frame_input.jpg', 'wraped.jpg', "output2.jpg"
 ####################################
 # Bonus: Creating GIFs of predicted flows
 # ---------------------------------------
@@ -259,4 +261,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(), gr.Image()]).launch()
+gr.Interface(fn=infer, inputs=[], outputs=[gr.Textbox(), gr.Image(), gr.Files(), gr.Image(), gr.Image(), gr.Image()]).launch()