Omnibus commited on
Commit
2e12e1a
·
1 Parent(s): f4ab656

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -81,7 +81,7 @@ def call_search(purpose, task, history, action_input):
81
  print (c)
82
  if c > MAX_DATA:
83
  print("compressing...")
84
- return_list = compress_data(c,return_list,task,history)
85
  history = "observation: the search results are:\n {}\n".format(return_list)
86
  else:
87
  history = "observation: I need to trigger a search using the following syntax:\naction: SEARCH action_input=SEARCH_QUERY\n"
@@ -104,7 +104,7 @@ def run_gpt(
104
 
105
  generate_kwargs = dict(
106
  temperature=0.9,
107
- max_new_tokens=1048,
108
  top_p=0.95,
109
  repetition_penalty=1.0,
110
  do_sample=True,
 
81
  print (c)
82
  if c > MAX_DATA:
83
  print("compressing...")
84
+ return_list = compress_data(c,purpose,task,return_list)
85
  history = "observation: the search results are:\n {}\n".format(return_list)
86
  else:
87
  history = "observation: I need to trigger a search using the following syntax:\naction: SEARCH action_input=SEARCH_QUERY\n"
 
104
 
105
  generate_kwargs = dict(
106
  temperature=0.9,
107
+ max_new_tokens=max_tokens,
108
  top_p=0.95,
109
  repetition_penalty=1.0,
110
  do_sample=True,