SmitaMahajan's picture
Update app.py
1eaca6d verified
raw
history blame contribute delete
719 Bytes
import gradio as gr
import joblib
# Load model and vectorizer
model = joblib.load("final_emotion_model.pkl")
vectorizer = joblib.load("final_vectorizer.pkl")
# Define accuracy (hardcoded or loaded from a file)
model_accuracy = 0.9974
# Define prediction function
def predict_emotion(text):
text_vec = vectorizer.transform([text])
prediction = model.predict(text_vec)[0]
return f"Prediction: {prediction}"
# Add accuracy info in the description or title
iface = gr.Interface(
fn=predict_emotion,
inputs="text",
outputs="text",
title="Emotional Manipulation Detector",
description=f"Detects manipulative intent in messages. Model accuracy: {model_accuracy*100:.2f}%"
)
iface.launch()