updates

app.py

@@ -1,7 +1,6 @@
 import streamlit as st
-from transformers import pipeline
-from pysentimiento import create_analyzer
-from transformers import AutoTokenizer, TFAutoModelForSequenceClassification
+import torch
+from transformers import AutoTokenizer, TFAutoModelForSequenceClassification, pipeline
 
 
 st.title("Toxic Tweets Analyzer")
@@ -13,7 +12,7 @@ st.image(image, use_column_width=True)
 model_name = st.selectbox("Select model", ["distilbert-base-uncased-finetuned-sst-2-english", "finiteautomata/bertweet-base-sentiment-analysis"])
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = TFAutoModelForSequenceClassification.from_pretrained(model_name)
-clf = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer)
+clf = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer, framework="pt")
 
 #form
 with st.form("my_form"):