Space launch

app.py

@@ -0,0 +1,35 @@
+import gradio as gr
+from transformers import AutoModelForSequenceClassification, AutoTokenizer
+import torch
+
+model_name = "ciro-c/emobot"
+model = AutoModelForSequenceClassification.from_pretrained(model_name)
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+
+mapping = {
+    3:'raiva',
+    4:'medo',
+    1:'alegria',
+    2:'amor',
+    0:'tristeza',
+    5:'surpresa'
+}
+
+def predict(text):
+    inputs = tokenizer(text, return_tensors="pt")
+
+    outputs = model(**inputs)
+    predictions = outputs.logits
+
+    return mapping[round(predictions.item())]
+
+iface = gr.Interface(
+    title="Identificador de emoções em uma frase",
+    description="Esse modelo foi treinado com o objetivo de identificar a emoção de uma frase escrita em inglês.  Ele identifica 6 emoções raiva, medo, alegria , amor, tristeza e surpresa "
+    fn=predict,
+    inputs="text",
+    layout="vertical",
+    outputs="text",
+)
+
+iface.launch(share=True)

requirements.txt

@@ -0,0 +1,3 @@
+gradio
+transformers
+torch