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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -30,7 +30,8 @@ def predict(prompt):
30
  }
31
 
32
  model = genai.GenerativeModel(
33
- model_name="gemini-2.0-flash-exp",
 
34
  generation_config=generation_config,
35
  )
36
 
@@ -39,8 +40,8 @@ def predict(prompt):
39
  ]
40
  )
41
 
42
- response = chat_session.send_message(prompt, tools='google_search_retrieval')
43
- #response = model.generate_content(contents=prompt, tools='google_search_retrieval')
44
  return response.text
45
 
46
  # Create the Gradio interface
 
30
  }
31
 
32
  model = genai.GenerativeModel(
33
+ model_name="gemini-1.5-pro",
34
+ #model_name="gemini-2.0-flash-exp",
35
  generation_config=generation_config,
36
  )
37
 
 
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