Spaces:
Runtime error
Runtime error
Commit
·
5911014
1
Parent(s):
eecfc2d
Update app.py
Browse files
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)
|