Update app.py
Browse files
app.py
CHANGED
|
@@ -174,5 +174,5 @@ def infer(frameA, frameB):
|
|
| 174 |
return "predicted_flow.jpg", ["flofile.flo"]
|
| 175 |
|
| 176 |
description="<p style='text-align:center'>PyTorch way to Generate optical flow image & .flo file from 2 consecutive frames with RAFT model</p>"
|
| 177 |
-
examples=[[
|
| 178 |
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")], title="RAFT Optical Flow", description=description, examples=examples).launch()
|
|
|
|
| 174 |
return "predicted_flow.jpg", ["flofile.flo"]
|
| 175 |
|
| 176 |
description="<p style='text-align:center'>PyTorch way to Generate optical flow image & .flo file from 2 consecutive frames with RAFT model</p>"
|
| 177 |
+
examples=[["./basket1.jpg","./basket2.jpg"]]
|
| 178 |
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")], title="RAFT Optical Flow", description=description, examples=examples).launch()
|