AmitJ82 commited on
Commit
59ec3e5
1 Parent(s): f5f3ef7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
10
 
11
  # Function to get response from Gemini model
12
  def get_gemini_response(input_prompt, image_data, genre, length, language, mood):
13
- model = genai.GenerativeModel('gemini-pro-vision')
14
  if image_data:
15
  full_prompt = f"""
16
  You are a famous creative writer. Look at the image provided and create a {length} {genre} in {language}. The {genre} should be {mood}. The {genre} should be based on the image and contain realistic and emotional elements.
 
10
 
11
  # Function to get response from Gemini model
12
  def get_gemini_response(input_prompt, image_data, genre, length, language, mood):
13
+ model = genai.GenerativeModel('gemini-1.5-flash')
14
  if image_data:
15
  full_prompt = f"""
16
  You are a famous creative writer. Look at the image provided and create a {length} {genre} in {language}. The {genre} should be {mood}. The {genre} should be based on the image and contain realistic and emotional elements.