fixing

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 import torch
+import numpy as np
 
 model_name = "dancrvlh/Language"
 model = AutoModelForSequenceClassification.from_pretrained(model_name)
@@ -19,7 +20,8 @@ def predict(text):
     inputs = tokenizer(text, return_tensors="pt")
     outputs = model(**inputs)
     predictions = outputs.logits
-    return mapping[round(predictions.item())]
+    return mapping[round(np.exp(predictions.item()))]
+    
 
 iface = gr.Interface(
     fn=predict,

requirements.txt

@@ -1,4 +1,5 @@
 pip
 gradio
 transformers
-torch
+torch
+numpy