shivambhosale commited on
Commit
6f4c263
1 Parent(s): 017d124

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -9,6 +9,9 @@ def getoutput(threshold,input_img):
9
  output_img = make_predictions(unet, input_img, threshold=threshold)
10
  return output_img
11
 
12
- demo = gr.Interface(getoutput, ["number",gr.Image(shape=(200, 200))], "image")
 
 
 
13
 
14
  demo.launch()
 
9
  output_img = make_predictions(unet, input_img, threshold=threshold)
10
  return output_img
11
 
12
+ demo = gr.Interface(
13
+ fn=getoutput,
14
+ inputs=["number", gr.Image(shape=(200, 200))],
15
+ outputs=["image"])
16
 
17
  demo.launch()