Gilvan commited on
Commit
553853a
·
verified ·
1 Parent(s): 1b6f5b4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ pipe.model.config.id2label = {0: 'greeting', 1: 'farewell', 2: 'other'}
19
  # Function to classify input text
20
  def classify_text(text):
21
  result = pipe(text)
22
- return result[0]['label']
23
 
24
  # Create Gradio interface
25
  iface = gr.Interface(
 
19
  # Function to classify input text
20
  def classify_text(text):
21
  result = pipe(text)
22
+ return result
23
 
24
  # Create Gradio interface
25
  iface = gr.Interface(