Spaces:

Gertie01
/

app-gb199hq1

Sleeping

Gertie01 commited on Nov 9

Commit

e03868b

verified ·

1 Parent(s): 554029a

Update models.py

Files changed (1) hide show

models.py CHANGED Viewed

@@ -71,7 +71,7 @@ def get_generation_prompt(
             # Contents should be images first, then the text prompt
             contents = images + [analysis_prompt]
             response = GEMINI_CLIENT.models.generate_content(
-                model='gemini-2.0-flash-live',
                 contents=contents
             )
             expanded_prompt = response.text.strip()

             # Contents should be images first, then the text prompt
             contents = images + [analysis_prompt]
             response = GEMINI_CLIENT.models.generate_content(
+                model='gemini-2.5-flash',
                 contents=contents
             )
             expanded_prompt = response.text.strip()