Update app.py

app.py

@@ -3,11 +3,11 @@ import gradio as gr
 import joblib
 
 # Configuration de base
-model_name = "facebook/bart-base"
-output_dir = "./bart_blagueur_model"
-# Tokenizer et modèle
-tokenizer = BartTokenizer.from_pretrained(model_name)
-model = BartForConditionalGeneration.from_pretrained(model_name)
+# model_name = "facebook/bart-base"
+# output_dir = "./bart_blagueur_model"
+# # Tokenizer et modèle
+# tokenizer = BartTokenizer.from_pretrained(model_name)
+# model = BartForConditionalGeneration.from_pretrained(model_name)
 model__ = joblib.load("model.joblib")
 tokenizer_ = joblib.load("tokenizer.joblib")
 
@@ -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,