Update app.py

app.py

@@ -9,10 +9,10 @@ import joblib
 # tokenizer = BartTokenizer.from_pretrained(model_name)
 # model = BartForConditionalGeneration.from_pretrained(model_name)
 model__ = joblib.load("model.joblib")
-tokenizer_ = joblib.load("tokenizer.joblib")
+tokenizer__ = joblib.load("tokenizer.joblib")
 
 def blagueur(prompt):
-    inputs = tokenizer_(prompt, return_tensors="pt", truncation=True, padding="max_length", max_length=128)
+    inputs = tokenizer__(prompt, return_tensors="pt", truncation=True, padding="max_length", max_length=128)
     outputs = model__.generate(
         input_ids=inputs["input_ids"],
         attention_mask=inputs["attention_mask"],
@@ -20,7 +20,7 @@ def blagueur(prompt):
         num_beams=5,
         do_sample=True,
         temperature=0.9)
-    return tokenizer_.decode(outputs[0], skip_special_tokens=True)
+    return tokenizer__.decode(outputs[0], skip_special_tokens=True)
 # Configuration de l'interface Gradio
 demo = gr.Interface(
     fn=blagueur,