updates

app.py

@@ -1,6 +1,6 @@
 import streamlit as st
 import torch
-from transformers import AutoTokenizer, TFAutoModelForSequenceClassification, pipeline
+from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline
 
 
 st.title("Toxic Tweets Analyzer")
@@ -11,7 +11,7 @@ st.image(image, use_column_width=True)
 #select model
 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)
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
 clf = pipeline("sentiment-analysis", model=model, tokenizer=tokenizer, framework="pt")
 
 #form