Update app.py

app.py

@@ -9,7 +9,7 @@ load_dotenv()
 
 # Configure Google Gemini API
 genai.configure(api_key=os.getenv('GOOGLE_API_KEY'))
-model = genai.GenerativeModel('gemini-2.0-flash')
+model = genai.GenerativeModel('gemini-pro')
 
 # Hide Streamlit menu and footer
 st.markdown("""
@@ -59,7 +59,11 @@ with col1:
                 and appeals to the target audience. Include a clear call to action."""
                 
                 try:
-                    response = model.generate_content(prompt, temperature=temperature)
+                    # Create generation config with temperature
+                    generation_config = genai.GenerationConfig(temperature=temperature)
+                    
+                    # Pass the generation config to generate_content
+                    response = model.generate_content(prompt, generation_config=generation_config)
                     st.success('✨ Your offer is ready!')
                     
                     # Display result in the right column