fix

app.py

@@ -100,7 +100,7 @@ def predict(s1, s2, chosen_model):
           with torch.no_grad():
               output = model(**model_input)
               logits = output[0][0].detach().numpy()
-              logits = softmax(logits).tolist()
+              score = logits[0]
               break
     def get_description(idx):
       description = score_descriptions[idx]
@@ -108,9 +108,12 @@ def predict(s1, s2, chosen_model):
       final_description = description + "\n \n" + description_pt
       return final_description
     
-    scores = { get_description(k):v for k,v in enumerate(logits) }
+    round_score = round(score)
+    score_percentage = min(score,5) / 5
+    score_percentage = '{:,.2%}'.format(score_percentage)
+    description = get_description(round_score).format(score_percentage)
 
-    return scores
+    return description
 
 
 inputs = [
@@ -120,7 +123,7 @@ inputs = [
 ]
 
 outputs = [
- gr.Label(label="Result")
+ gr.Textbox(label="Result")
 ]