changes

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
+import numpy as np
 
 model_name = "Bittar/outputs"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
@@ -17,15 +18,13 @@ def predict(text):
     outputs = model(**inputs)
     predictions = outputs.logits
 
-    return mapping[round(predictions.item())]
+    return mapping[predictions.argmax()]
 
 iface = gr.Interface(
     fn=predict,
     inputs="text",
     outputs="text",
-    layout="vertical",
-    title="Classificador de emoções em uma frase",
-    description="Este modelo analisa uma frase em inglês e diz qual sentimento mais se aproxima da frase apresentada. A frase pode ser classificada em Joy, Anger e Fear"
+    layout="vertical"
 )
 
 iface.launch(share=True)

requirements.txt

@@ -1,3 +1,4 @@
 gradio
 transformers
-torch
+torch
+numpy