fffiloni commited on
Commit
c00946c
1 Parent(s): 2a59082

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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()