Spaces:

Nechba
/

seo_article_generate

Runtime error

Nechba commited on Jul 17, 2024

Commit

5413f58

verified ·

1 Parent(s): b20cc25

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -8,7 +8,7 @@ def call_ai_api(prompt):
     "top_p": 0.95,
     "max_output_tokens": 5000000,
     }
-    model = genai.GenerativeModel(model_name="gemini-1.5-pro-latest",
                                 generation_config=generation_config)
     response = model.generate_content(prompt)
     return response.text

     "top_p": 0.95,
     "max_output_tokens": 5000000,
     }
+    model = genai.GenerativeModel(model_name="gemini-1.0-pro-latest",
                                 generation_config=generation_config)
     response = model.generate_content(prompt)
     return response.text