Mori-kamiyama commited on
Commit
3611e4e
·
1 Parent(s): 5360051

Merge remote-tracking branch 'origin/main'

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ index.add(np.array(doc_embeddings))
48
 
49
  def generate_text(prompt):
50
  full_prompt = search(prompt)
51
- output = llm(full_prompt, max_tokens=256)
52
  return output["choices"][0]["text"]
53
 
54
  def search(query):
 
48
 
49
  def generate_text(prompt):
50
  full_prompt = search(prompt)
51
+ output = llm(full_prompt, max_tokens=128)
52
  return output["choices"][0]["text"]
53
 
54
  def search(query):