defaults

app.py

@@ -39,6 +39,12 @@ displaying the relationship between the text pairs.
 exibindo a relação entre os pares textuais.)
 """ }
 
+short_score_descriptions = {
+  1: "Entailment (Implicação)",
+  0: "No relationship (Nenhuma relação)",
+  2: "Paraphrase (Paráfrase)"
+}
+
 score_descriptions = {
   1: "There is an entailment relation between premise and hypothesis. If the premise is true, then the hypothesis must also be true.",
   0: "There is no logical relation between the premise and the hypothesis.",
@@ -116,19 +122,22 @@ 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) }
+    max_pos = logits.index(max(logits))
+    description = get_description(max_pos)
+    scores = { short_score_descriptions[k]:v for k,v in enumerate(logits) }
 
-    return scores
+    return scores, description
 
 
 inputs = [
-    gr.inputs.Textbox(label="Premise"),
-    gr.inputs.Textbox(label="Hypothesis"),
-    gr.Dropdown(label="Model", choices=user_friendly_name_list, default=user_friendly_name_list[0])
+    gr.Textbox(label="Premise", value=app_examples[0][0]),
+    gr.Textbox(label="Hypothesis", value=app_examples[0][1]),
+    gr.Dropdown(label="Model", choices=user_friendly_name_list, value=user_friendly_name_list[0])
 ]
 
 outputs = [
- gr.Label(label="Result")
+ gr.Label(label="Result"),
+ gr.Markdown()
 ]