Loren commited on
Commit
4a84b1d
·
verified ·
1 Parent(s): dfbd9e3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -210,7 +210,7 @@ def get_answer_with_query(query, use_rerank, history):
210
  # Construction du prompt
211
  prompt = RAG_PROMPT_TEMPLATE.format(
212
  context="\n".join(list_chunks),
213
- question=user_query
214
  )
215
  # Génération de la réponse
216
  inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
 
210
  # Construction du prompt
211
  prompt = RAG_PROMPT_TEMPLATE.format(
212
  context="\n".join(list_chunks),
213
+ question=query
214
  )
215
  # Génération de la réponse
216
  inputs = tokenizer(prompt, return_tensors="pt").to(model.device)