ameerazam08 commited on
Commit
bcbcb94
·
verified ·
1 Parent(s): 3093758

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -77,7 +77,7 @@ def bot(
77
  response.resolve()
78
  else:
79
  image_prompt = preprocess_image(image_prompt)
80
- model = genai.GenerativeModel('gemini-pro-vision')
81
  response = model.generate_content(
82
  contents=[text_prompt, image_prompt],
83
  stream=True,
 
77
  response.resolve()
78
  else:
79
  image_prompt = preprocess_image(image_prompt)
80
+ model = genai.GenerativeModel(model_name)
81
  response = model.generate_content(
82
  contents=[text_prompt, image_prompt],
83
  stream=True,