kietnt0603 commited on
Commit
54d79e1
1 Parent(s): 02e5658

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ article = "Model loaded from https://huggingface.co/kietnt0603/bertweet-base-hat
37
  # Create the Gradio interface
38
  iface = gr.Interface(fn=predict,
39
  inputs="textbox",
40
- outputs="dictionary",
41
  title=title,
42
  description=description,
43
  article=article)
 
37
  # Create the Gradio interface
38
  iface = gr.Interface(fn=predict,
39
  inputs="textbox",
40
+ outputs=gr.Label(num_top_classes=3, label='Predictions'),
41
  title=title,
42
  description=description,
43
  article=article)