Update app.py

app.py

@@ -6,8 +6,6 @@ import random as r
 import asyncio
 import gradio as gr
 
-gr.Interface.load("models/APJ23/MultiHeaded_Sentiment_Analysis_Model").launch()
-
 tokenizer = AutoTokenizer.from_pretrained("APJ23/MultiHeaded_Sentiment_Analysis_Model")
 model = AutoModelForSequenceClassification.from_pretrained("APJ23/MultiHeaded_Sentiment_Analysis_Model")
 
@@ -20,8 +18,9 @@ classes = {
     5: 'Insult',
     6: 'Identity Hate'
 }
+
 @st.cache(allow_output_mutation=True)
-def prediction(tweet,model,tokenizer):
+def prediction(tweet, model, tokenizer):
     inputs = tokenizer(tweet, return_tensors="pt", padding=True, truncation=True)
     outputs = model(**inputs)
     predicted_class = torch.argmax(outputs.logits, dim=1)
@@ -38,11 +37,10 @@ def create_table(predictions):
     return df
 
 st.title('Toxicity Prediction App')
-tweet=st.text_input('Enter a tweet to check for toxicity')
-
+tweet = st.text_input('Enter a tweet to check for toxicity')
 
 if st.button('Predict'):
-    predicted_class_label, predicted_prob = prediction(tweet, model, tokenizer))
+    predicted_class_label, predicted_prob = prediction(tweet, model, tokenizer)
     prediction_text = f'Prediction: {predicted_class_label} ({predicted_prob:.2f})'
     st.write(prediction_text)
     predictions = {tweet: (predicted_class_label, predicted_prob)}