e1010101 commited on
Commit
f498967
1 Parent(s): ad7d1cf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -31,8 +31,11 @@ def classify_image(image):
31
 
32
  interface = gr.Interface(
33
  fn=classify_image,
34
- inputs=gr.inputs.Image(type="pil"),
35
- outputs=gr.outputs.Label(num_top_classes=None),
 
 
 
36
  title="Multi-Label Image Classification",
37
  description="Upload an image to get classification results."
38
  )
 
31
 
32
  interface = gr.Interface(
33
  fn=classify_image,
34
+ inputs=[
35
+ gr.Image(type="pil"),
36
+ gr.Slider(minimum=0, maximum=1, value=0.5, label="Probability Threshold")
37
+ ],
38
+ outputs=gr.Label(num_top_classes=None),
39
  title="Multi-Label Image Classification",
40
  description="Upload an image to get classification results."
41
  )