Update app.py

app.py

@@ -10,6 +10,24 @@ def interpret_pred(pred):
     is_ai_percent = round(100 * interpreted_pred)
     return result, is_ai_percent
 
+def interpret_pred_with_sensibility(pred):
+    low_bond = -6.748472
+    high_bound = 6.7176056
+    pred_value = pred[0][1].item()
+    interpreted_pred = (pred_value - low_bond) / (high_bound - low_bond)
+    if interpreted_pred < 0.5:
+        proba = "très faible"
+    elif interpreted_pred < 0.6:
+        proba = "faible"
+    elif interpreted_pred < 0.8:
+        proba = "modérée"
+    elif interpreted_pred < 0.95:
+        proba = "élevée"
+    else:
+        proba = "très élevée"
+
+    return proba
+
 
 def main(text_sentence):
     import torch
@@ -25,11 +43,16 @@ def main(text_sentence):
 
     predict = model.forward(input_ids)[0]
 
+    #result = (
+    #    "Résultat : {}.\nCe texte a {}% de chances d'avoir été généré par de l'IA"
+    #    .format(interpret_pred(predict)[0], interpret_pred(predict)[1])
+    #)
+
+    proba = interpret_pred_with_sensibility(predict)
     result = (
-        "Résultat : {}.\nCe texte a {}% de chances d'avoir été généré par de l'IA"
-        .format(interpret_pred(predict)[0], interpret_pred(predict)[1])
+        "La probabilité que ce texte a été généré par de l'IA est {}"
+        .format(proba)
     )
-
     return result