mgokg commited on
Commit
2efb0ef
·
verified ·
1 Parent(s): 5b85257

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,7 +22,7 @@ custom_css = """
22
  def predict(prompt):
23
  # Create the model
24
  generation_config = {
25
- "temperature": 1,
26
  "top_p": 0.95,
27
  "top_k": 40,
28
  "max_output_tokens": 2048,
@@ -40,8 +40,8 @@ def predict(prompt):
40
  ]
41
  )
42
 
43
- #response = chat_session.send_message(prompt)
44
- response = model.generate_content(contents=prompt, tools='google_search_retrieval')
45
  return response.text
46
 
47
  # Create the Gradio interface
 
22
  def predict(prompt):
23
  # Create the model
24
  generation_config = {
25
+ "temperature": 0.3,
26
  "top_p": 0.95,
27
  "top_k": 40,
28
  "max_output_tokens": 2048,
 
40
  ]
41
  )
42
 
43
+ response = chat_session.send_message(prompt)
44
+ #response = model.generate_content(contents=prompt, tools='google_search_retrieval')
45
  return response.text
46
 
47
  # Create the Gradio interface