sitammeur commited on
Commit
19ae9d3
·
verified ·
1 Parent(s): bfd2f4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,4 +1,7 @@
1
  # Necessary imports
 
 
 
2
  import gradio as gr
3
  from src.siglip.classifier import ZeroShotImageClassification
4
 
@@ -35,7 +38,7 @@ demo = gr.Interface(
35
  placeholder="Enter candidate labels separated by commas",
36
  ),
37
  ],
38
- outputs=gr.Label(label="Classification"),
39
  title=title,
40
  description=description,
41
  article=article,
 
1
  # Necessary imports
2
+ import warnings
3
+ warnings.filterwarnings("ignore")
4
+
5
  import gradio as gr
6
  from src.siglip.classifier import ZeroShotImageClassification
7
 
 
38
  placeholder="Enter candidate labels separated by commas",
39
  ),
40
  ],
41
+ outputs=gr.Label(label="Classification", num_top_classes=3),
42
  title=title,
43
  description=description,
44
  article=article,