Update app.py

app.py

@@ -171,7 +171,7 @@ def infer(frameA, frameB):
     
     flo_file = write_flo(predicted_flow, "flofile.flo")
     
-    return "done", "predicted_flow.jpg", ["flofile.flo"]
+    return "predicted_flow.jpg", ["flofile.flo"]
 
       
-gr.Interface(fn=infer, inputs=[gr.Image(source="upload", type="filepath", label="frame 1"), gr.Image(source="upload", type="filepath", label="frame 2")], outputs=[gr.Textbox(), gr.Image(label="flow"), gr.Files()]).launch()
+gr.Interface(fn=infer, inputs=[gr.Image(source="upload", type="filepath", label="frame 1"), gr.Image(source="upload", type="filepath", label="frame 2")], outputs=[gr.Image(label="flow image"), gr.Files(label="flow file")]).launch()