mgokg commited on
Commit
5878e17
·
verified ·
1 Parent(s): 54f37fd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -40,7 +40,8 @@ def predict(prompt):
40
  generation_config=generation_config,
41
  )
42
 
43
- contents_to_send = [genai.Content(parts=[prompt])]
 
44
  response = model.generate_content(contents=contents_to_send, tools='google_search_retrieval')
45
  return response.text
46
 
 
40
  generation_config=generation_config,
41
  )
42
 
43
+ #contents_to_send = [genai.Content(parts=[prompt])]
44
+ contents_to_send = [prompt]
45
  response = model.generate_content(contents=contents_to_send, tools='google_search_retrieval')
46
  return response.text
47