Update app.py

app.py

@@ -67,7 +67,7 @@ def write_flo(flow, filename):
 
 
     
-def infer():
+def infer(frameA, frameB):
     #video_url = "https://download.pytorch.org/tutorial/pexelscom_pavel_danilyuk_basketball_hd.mp4"
     #video_path = Path(tempfile.mkdtemp()) / "basketball.mp4"
     #_ = urlretrieve(video_url, video_path)
@@ -76,9 +76,9 @@ def infer():
     #print(f"FRAME BEFORE stack: {frames[100]}")
     
     
-    input_frame_1 = read_image(str("./basket1.jpg"), ImageReadMode.UNCHANGED)
+    input_frame_1 = read_image(str(frameA), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_1}")
-    input_frame_2 = read_image(str("./basket2.jpg"), ImageReadMode.UNCHANGED)
+    input_frame_2 = read_image(str(frameB), ImageReadMode.UNCHANGED)
     print(f"FRAME 1: {input_frame_2}")
     
     #img1_batch = torch.stack([frames[0]])
@@ -174,4 +174,4 @@ def infer():
     return "done", "predicted_flow.jpg", ["flofile.flo"]
 
       
-gr.Interface(fn=infer, inputs=[], 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.Textbox(), gr.Image(label="flow"), gr.Files()]).launch()