Spaces:

hruday96
/

prompt-lab

Running

hruday96 commited on Mar 9

Commit

bcdf45a

verified ·

1 Parent(s): b6e5361

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -68,7 +68,7 @@ if st.button("Generate Enhanced Prompt"):
                 prompt = RAIKAGE_TEMPLATE.format(user_prompt=topic)
             # Initialize the generative model
-            model = genai.GenerativeModel('gemini-pro')
             # Generate enhanced prompt
             try:

                 prompt = RAIKAGE_TEMPLATE.format(user_prompt=topic)
             # Initialize the generative model
+            model = genai.GenerativeModel('gemini-2.0-flash')
             # Generate enhanced prompt
             try: