neverix commited on
Commit
2c160aa
·
1 Parent(s): 1750ab0

Add features

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ class MidasDepth(object):
29
 
30
  def main():
31
  midas = MidasDepth()
32
- interface = gr.Interface(inputs=[
33
  gr.inputs.Image(),
34
  gr.inputs.Text()
35
  ], outputs=[
 
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.Text()
35
  ], outputs=[