Update app.py

app.py

@@ -171,7 +171,7 @@ def infer(frameA, frameB):
     
     flo_file = write_flo(predicted_flow, "flofile.flo")
     
-    return "predicted_flow.jpg", ["flofile.flo"]
+    return "predicted_flow.jpg", "flofile.flo"
 
 title="RAFT Optical Flow"
 description="<p style='text-align:center'>PyTorch way to Generate optical flow image & .flo file from 2 consecutive frames with RAFT model</p>" 
@@ -187,7 +187,7 @@ with gr.Blocks() as block:
         submit_btn = gr.Button("Submit")
         
         flow_img_out = gr.Image(label="flow image") 
-        flow_file_out = gr.Files(label="flow file")
+        flow_file_out = gr.File(label="flow file")
         
         
         examples=[['basket1.jpg','basket2.jpg']]