Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 (
|
| 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)
|