Update app.py

app.py

@@ -33,10 +33,10 @@ def interpret_pred_with_sensibility(pred):
     return proba
 
 
-def main(text_sentence):
+def main(Texte -> Resultat):
 
     input_ids = torch.tensor(
-        [barthez_tokenizer.encode(text_sentence, truncation=True, padding=True, add_special_tokens=True)]
+        [barthez_tokenizer.encode(Texte, truncation=True, padding=True, add_special_tokens=True)]
     )
 
     predict = model.forward(input_ids)[0]
@@ -47,11 +47,11 @@ def main(text_sentence):
     #)
 
     proba = interpret_pred_with_sensibility(predict)
-    result = (
+    Resultat = (
         "La probabilité que ce texte a été généré par de l'IA est {}"
         .format(proba)
     )
-    return result
+    return Resultat
 
 
 iface = gr.Interface(fn=main, inputs="text", outputs="text")