Spaces:

hruday96
/

prompt-lab

Running

hruday96 commited on Mar 9

Commit

6d829d8

verified ·

1 Parent(s): 8416685

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -67,7 +67,7 @@ if st.button("🚀 Enhance Prompt"):
             # ✅ Call Gemini API to Enhance the Prompt
             try:
-                model = genai.GenerativeModel('gemini-pro')
                 response = model.generate_content(full_prompt)
                 # ✅ Display Enhanced Prompt

             # ✅ Call Gemini API to Enhance the Prompt
             try:
+                model = genai.GenerativeModel('gemini-2.0-flash')
                 response = model.generate_content(full_prompt)
                 # ✅ Display Enhanced Prompt