Update app.py
Browse files
app.py
CHANGED
@@ -173,5 +173,5 @@ def infer(frameA, frameB):
|
|
173 |
|
174 |
return "predicted_flow.jpg", ["flofile.flo"]
|
175 |
|
176 |
-
|
177 |
-
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()
|
|
|
173 |
|
174 |
return "predicted_flow.jpg", ["flofile.flo"]
|
175 |
|
176 |
+
description="Generate flow image & .flo file from 2 frames with RAFT model"
|
177 |
+
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 Flow generation", description=description).launch()
|