JethroDD commited on
Commit
679edd6
1 Parent(s): 73d4b43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ from transformers import TrainingArguments, Trainer
10
 
11
  def sentiment(comment):
12
 
13
- model = pipeline("text-classification", model="cardiffnlp/twitter-roberta-base-sentiment-latest")
14
  outputs = model(comment)
15
  result_text =outputs[0]["label"]
16
  print(f"The predicted class is {result_text}")
 
10
 
11
  def sentiment(comment):
12
 
13
+ model = pipeline("text-classification", model="JethroDD/twitter-roberta-base-sentiment-CustomModel_imdb")
14
  outputs = model(comment)
15
  result_text =outputs[0]["label"]
16
  print(f"The predicted class is {result_text}")