Update app.py

app.py

@@ -9,9 +9,13 @@ if "history" not in st.session_state:
     st.session_state.history = []
     
 def get_models():
-  tokenizer = BlenderbotTokenizer.from_pretrained("facebook/blenderbot-400M-distill")
-  model = BlenderbotForConditionalGeneration.from_pretrained("facebook/blenderbot-400M-distill")
-  return tokenizer, model
+    
+    model_name = "tbboukhari/chatbot-produit-fr"
+    
+    model = AutoModelForSeq2SeqLM.from_pretrained(model_name, device_map="auto")
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+ 
+    return tokenizer, model
   
 def generate_answer():
     tokenizer, model = get_models()