Spaces:
Runtime error
Runtime error
File size: 2,195 Bytes
ab4de6e 6714a50 9e3035c 3c380c2 9e3035c 197d2f5 99abe4d eb9f908 ab4de6e 3c380c2 ab4de6e 3c380c2 4894de4 3c380c2 ab4de6e 3c380c2 ab4de6e 3c380c2 ab4de6e 3c380c2 ab4de6e 3c380c2 ab4de6e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 |
import streamlit as st
import pandas as pd
import torch
import asyncio
import gradio as gr
from transformers import pipeline, AutoTokenizer, AutoModelForSequenceClassification
gr.Interface.load("models/APJ23/MultiHeaded_Sentiment_Analysis_Model").launch()
tokenizer = AutoTokenizer.from_pretrained("APJ23/MultiHeaded_Sentiment_Analysis_Model", local_files_only=True)
model = AutoModelForSequenceClassification.from_pretrained("APJ23/MultiHeaded_Sentiment_Analysis_Model")
classes = {
0: 'Non-Toxic',
1: 'Toxic',
2: 'Severely Toxic',
3: 'Obscene',
4: 'Threat',
5: 'Insult',
6: 'Identity Hate'
}
@st.cache(allow_output_mutation=True)
def predict_toxicity(tweet, model, tokenizer):
inputs = tokenizer(tweet, return_tensors="pt", padding=True, truncation=True)
outputs = model(**inputs)
predicted_class = torch.argmax(outputs.logits, dim=1)
predicted_prob = torch.softmax(outputs.logits, dim=1)[0][predicted_class].item()
return classes[predicted_class], predicted_prob
def create_table(predictions):
data = {'Tweet': [], 'Highest Toxicity Class': [], 'Probability': []}
for tweet, prediction in predictions.items():
data['Tweet'].append(tweet)
data['Highest Toxicity Class'].append(prediction[0])
data['Probability'].append(prediction[1])
df = pd.DataFrame(data)
return df
async def run_async_prediction(tweet, model, tokenizer):
loop = asyncio.get_event_loop()
prediction = await loop.run_in_executor(None, predict_toxicity, tweet, model, tokenizer)
return prediction
st.title('Toxicity Prediction App')
tweet_input = st.text_input('Enter a tweet to check for toxicity')
if st.button('Predict'):
predictions = {tweet_input: None}
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
prediction = loop.run_until_complete(run_async_prediction(tweet_input, model, tokenizer))
predictions[tweet_input] = prediction
loop.close()
predicted_class_label, predicted_prob = prediction
prediction_text = f'Prediction: {predicted_class_label} ({predicted_prob:.2f})'
st.write(prediction_text)
table = create_table(predictions)
st.table(table)
|