cccc commited on
Commit
98dd801
·
1 Parent(s): f971353

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def sentiment_analysis(sentence, template, model_name, positive, neutral, negati
65
  demo = gr.Interface(fn=sentiment_analysis,
66
  inputs = [gr.Textbox(placeholder="Enter sentence here.",label="sentence"),
67
  gr.Textbox(placeholder="Your template must have a [SENTENCE] token and a [MASK] token.",label="template"),
68
- gr.Radio(choices=["ARCH_tuned_RoBerta","FNCH_tuned_RoBerta","AREN_tuned_RoBerta","FNEN_tuned_RoBerta","bert-base-uncased"], label="model choics"),
69
  gr.Textbox(placeholder="Separate words with Spaces.",label="positive label words"),
70
  gr.Textbox(placeholder="Separate words with Spaces.",label="neutral label words"),
71
  gr.Textbox(placeholder="Separate words with Spaces.",label="negative label words")
 
65
  demo = gr.Interface(fn=sentiment_analysis,
66
  inputs = [gr.Textbox(placeholder="Enter sentence here.",label="sentence"),
67
  gr.Textbox(placeholder="Your template must have a [SENTENCE] token and a [MASK] token.",label="template"),
68
+ gr.Radio(choices=["roberta-base","bert-base-uncased","yiyanghkust/finbert-pretrain"], label="model choics"),
69
  gr.Textbox(placeholder="Separate words with Spaces.",label="positive label words"),
70
  gr.Textbox(placeholder="Separate words with Spaces.",label="neutral label words"),
71
  gr.Textbox(placeholder="Separate words with Spaces.",label="negative label words")