Update app.py

app.py

@@ -34,7 +34,7 @@ if st.button("Generate"):
     if medicine_name:  # Ensure the medicine name is entered
         try:
             # Initialize the generative model (adjust model name if needed)
-            model = genai.GenerativeModel('gemini-pro')  # Ensure this is the correct model name
+            model = genai.GenerativeModel('gemini-2.0-flash')  # Ensure this is the correct model name
 
             # Generate content based on the prompt
             response = model.generate_content(prompt)