Update app.py

app.py

@@ -4,6 +4,7 @@ import torch
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import random as r
 import gradio as gr
+import asyncio
 
 gr.Interface.load("models/APJ23/MultiHeaded_Sentiment_Analysis_Model").launch()
 
@@ -38,6 +39,9 @@ def create_table(predictions):
 
 st.title('Toxicity Prediction App')
 tweet=st.text_input('Enter a tweet to check for toxicity')
+async def run_async_function():
+    result = await prediction(tweet, model, tokenizer)
+    return result
 if st.button('Predict'):
     predicted_class_label, predicted_prob = predict_toxicity(tweet, model, tokenizer)
     prediction_text = f'Prediction: {predicted_class_label} ({predicted_prob:.2f})'