edit

app.py

@@ -1,11 +1,11 @@
 import numpy as np
 import streamlit as st
 from transformers import pipeline
-from transformers import AutoTokenizer, AutoModelForSequenceClassification
+
 import torch
 
 def bertweet(data):
-    specific_model = AutoModelForSequenceClassification.from_pretrained("finiteautomata/bertweet-base-sentiment-analysis")
+    specific_model = pipeline(model="finiteautomata/bertweet-base-sentiment-analysis")
     result = specific_model(data)
     label = result[0]['label']
     score = result[0]['score']
@@ -13,7 +13,7 @@ def bertweet(data):
     return label, score 
 
 def roberta(data):
-    specific_model = AutoModelForSequenceClassification.from_pretrained("cardiffnlp/twitter-roberta-base-sentiment")
+    specific_model = pipeline(model="cardiffnlp/twitter-roberta-base-sentiment")
     result = specific_model(data)
     label = result[0]['label']
     score = result[0]['score']