sitammeur commited on
Commit
099ab82
·
verified ·
1 Parent(s): 4b91f1f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -5
app.py CHANGED
@@ -1,9 +1,6 @@
1
  # Necessary imports
2
- import warnings
3
- warnings.filterwarnings("ignore")
4
-
5
  import gradio as gr
6
- from src.app.predict import ZeroShotTextClassification
7
 
8
 
9
  # Examples to display in the interface
@@ -35,7 +32,7 @@ article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2412.136
35
  demo = gr.Interface(
36
  fn=ZeroShotTextClassification,
37
  inputs=[
38
- gr.Textbox(label="Text", placeholder="Enter text to classify"),
39
  gr.Textbox(
40
  label="Candidate Labels",
41
  placeholder="Enter candidate labels separated by commas",
 
1
  # Necessary imports
 
 
 
2
  import gradio as gr
3
+ from classytext.classifier.predict import ZeroShotTextClassification
4
 
5
 
6
  # Examples to display in the interface
 
32
  demo = gr.Interface(
33
  fn=ZeroShotTextClassification,
34
  inputs=[
35
+ gr.Textbox(label="Input", placeholder="Enter text to classify"),
36
  gr.Textbox(
37
  label="Candidate Labels",
38
  placeholder="Enter candidate labels separated by commas",