Satyam-Singh commited on
Commit
05c6253
·
verified ·
1 Parent(s): 0604e71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ genai.configure(
7
  api_key=os.getenv("api_key")
8
  )
9
 
10
- models = "gemini-pro"
11
 
12
  def generate_text(prompt, examples, model=models, temperature=0.25):
13
  return genai.chat(model=model, temperature=temperature, examples=examples, messages=prompt).last
 
7
  api_key=os.getenv("api_key")
8
  )
9
 
10
+ models = "models/gemini-pro"
11
 
12
  def generate_text(prompt, examples, model=models, temperature=0.25):
13
  return genai.chat(model=model, temperature=temperature, examples=examples, messages=prompt).last