chansung commited on
Commit
c92a273
1 Parent(s): d433cb4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -72,15 +72,14 @@ def internet_search(ppmanager, serper_api_key, global_context, ctx_num_lconv, de
72
  internet_search_ppm.pingpongs[-1].ping = internet_search_prompt
73
  internet_search_prompt = build_prompts(internet_search_ppm, "", win_size=ctx_num_lconv)
74
 
75
- instruction = gen_text_none_stream(internet_search_prompt, hf_model=MODEL_ID, hf_token=TOKEN)
76
  ###
77
 
78
  searcher = SimilaritySearcher.from_pretrained(device=device)
79
  iss = InternetSearchStrategy(
80
- searcher,
81
- instruction=instruction,
82
  serper_api_key=serper_api_key
83
- )(ppmanager)
84
 
85
  step_ppm = None
86
  while True:
 
72
  internet_search_ppm.pingpongs[-1].ping = internet_search_prompt
73
  internet_search_prompt = build_prompts(internet_search_ppm, "", win_size=ctx_num_lconv)
74
 
75
+ search_query = gen_text_none_stream(internet_search_prompt, hf_model=MODEL_ID, hf_token=TOKEN)
76
  ###
77
 
78
  searcher = SimilaritySearcher.from_pretrained(device=device)
79
  iss = InternetSearchStrategy(
80
+ searcher,
 
81
  serper_api_key=serper_api_key
82
+ )(ppmanager, search_query=search_query)
83
 
84
  step_ppm = None
85
  while True: