TropicalBee's picture
adding ssr_mode
3113b06 verified
Raw
History Blame Contribute Delete
519 Bytes
import gradio as gr
from transformers import pipeline
model = pipeline(
"text-classification",
model="TropicalBee/talk2self-emotion-detection"
)
def predict(text):
results = model(text, top_k=None)
return {r["label"]: r["score"] for r in results}
demo = gr.Interface(
fn=predict,
inputs=gr.Textbox(placeholder="Type how you're feeling..."),
outputs=gr.Label(num_top_classes=7),
title="Talk2Self Emotion Detection"
)
demo.launch(server_name="0.0.0.0", server_port=7860,ssr_mode=False)