Fixing result error

app.py

@@ -20,8 +20,10 @@ def predict(text):
 
     outputs = model(**inputs)
     predictions = outputs.logits
-
-    return mapping[round(predictions.item())]
+    result = predictions.item()
+    result = max(result, 0)
+    result = min(result, 5)
+    return mapping[round(result)]
 
 iface = gr.Interface(
     title="Identificador de emoções em uma frase",