xavierbarbier commited on
Commit
2aa1606
·
verified ·
1 Parent(s): 260dc47

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -65,7 +65,7 @@ def qa(question: str) -> str:
65
  docs = vectordb.max_marginal_relevance_search(question,k=1)
66
 
67
 
68
-
69
  prompt = f"""<s>[INST]
70
  Les informations contextuelles sont ci-dessous.
71
  ---------------------
@@ -76,8 +76,9 @@ def qa(question: str) -> str:
76
  [INST] Requête: {question} [/INST]
77
  Réponse:
78
  """
 
79
  #outputs = model.generate(prompt=prompt, temp=0.5, top_k = 40, top_p = 1, max_tokens = max_new_tokens)
80
- return prompt #"".join(outputs)
81
 
82
 
83
  demo = gr.Interface(
 
65
  docs = vectordb.max_marginal_relevance_search(question,k=1)
66
 
67
 
68
+ """
69
  prompt = f"""<s>[INST]
70
  Les informations contextuelles sont ci-dessous.
71
  ---------------------
 
76
  [INST] Requête: {question} [/INST]
77
  Réponse:
78
  """
79
+ """
80
  #outputs = model.generate(prompt=prompt, temp=0.5, top_k = 40, top_p = 1, max_tokens = max_new_tokens)
81
+ return docs[0].page_content #"".join(outputs)
82
 
83
 
84
  demo = gr.Interface(