neverix commited on
Commit
a336e88
·
1 Parent(s): 279f60d

Add everything

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -29,9 +29,14 @@ class MidasDepth(object):
29
 
30
  def main():
31
  midas = MidasDepth()
32
- interface = gr.Interface(fn=lambda x: [Image.fromarray(midas.get_depth(x[0]).astype("uint8")), ""], inputs=[
33
  gr.inputs.Image(),
34
- gr.inputs.Textbox()
 
 
 
 
 
35
  ], outputs=[
36
  gr.outputs.Image(),
37
  gr.outputs.Video()
 
29
 
30
  def main():
31
  midas = MidasDepth()
32
+ interface = gr.Interface(fn=lambda x, *t: [Image.fromarray(midas.get_depth(x[0]).astype("uint8")), ""], inputs=[
33
  gr.inputs.Image(),
34
+ gr.inputs.Number(label="tx", default=0.0),
35
+ gr.inputs.Number(label="ty", default=0.0),
36
+ gr.inputs.Number(label="tz", default=0.0),
37
+ gr.inputs.Number(label="rx", default=0.0),
38
+ gr.inputs.Number(label="ry", default=0.0),
39
+ gr.inputs.Number(label="rz", default=0.0)
40
  ], outputs=[
41
  gr.outputs.Image(),
42
  gr.outputs.Video()