ningrumdaud commited on
Commit
74cf769
·
verified ·
1 Parent(s): 543baf6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ model2 = pipeline(model="finiteautomata/bertweet-base-sentiment-analysis")
7
 
8
  def predict_sentiment(text, model_choice):
9
  try:
10
- if model_choice == "Model 1 (Default Transformer)":
11
  predictions = model1(text)
12
  elif model_choice == "Model 2 (BERTweet)":
13
  predictions = model2(text)
 
7
 
8
  def predict_sentiment(text, model_choice):
9
  try:
10
+ if model_choice == "Model 1 (RoBERTa-large)":
11
  predictions = model1(text)
12
  elif model_choice == "Model 2 (BERTweet)":
13
  predictions = model2(text)