tbboukhari commited on
Commit
5911014
·
1 Parent(s): eecfc2d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -9,6 +9,7 @@ st.title("Chatbot Produit")
9
 
10
  if "history" not in st.session_state:
11
  st.session_state.history = []
 
12
 
13
  def get_models():
14
 
@@ -22,9 +23,9 @@ def get_models():
22
  model = PeftModel.from_pretrained(model, peft_model_id)
23
 
24
  return tokenizer, model
25
-
 
26
  def generate_answer():
27
- tokenizer, model = get_models()
28
  user_message = st.session_state.input_text
29
  inputs = tokenizer(st.session_state.input_text, return_tensors="pt")
30
  result = model.generate(**inputs)
 
9
 
10
  if "history" not in st.session_state:
11
  st.session_state.history = []
12
+ st.session_state.history.append({"message": "Bonjour ! Donnez moi une phrase et j'essaierai de trouver le domaine associé", "is_user": False})
13
 
14
  def get_models():
15
 
 
23
  model = PeftModel.from_pretrained(model, peft_model_id)
24
 
25
  return tokenizer, model
26
+
27
+ tokenizer, model = get_models()
28
  def generate_answer():
 
29
  user_message = st.session_state.input_text
30
  inputs = tokenizer(st.session_state.input_text, return_tensors="pt")
31
  result = model.generate(**inputs)