fffiloni commited on
Commit
d4c86a0
1 Parent(s): 5671479

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -1
app.py CHANGED
@@ -175,4 +175,14 @@ def infer(frameA, frameB):
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
 
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).launch()
 
 
 
 
 
 
 
 
 
 
 
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
 
178
+ gr.Interface(fn=infer,
179
+ inputs=[
180
+ gr.Image(source="upload", type="filepath", label="frame 1"),
181
+ gr.Image(source="upload", type="filepath", label="frame 2")],
182
+ outputs=[
183
+ gr.Image(label="flow image"),
184
+ gr.Files(label="flow file")],
185
+ title="RAFT Optical Flow",
186
+ description=description,
187
+ examples=[["basket1.jpg", "basket2.jpg"]]
188
+ ).launch()