Update app.py

app.py

@@ -50,7 +50,8 @@ def main():
             response = generate_gemini_content(prompt, model_name=model_name)
 
         # Display the generated content in Markdown format
-        st.markdown(to_markdown(response.text))
+        generated_text = response.candidates[0].content.parts[0].text
+        st.markdown(to_markdown(generated_text))
 
 if __name__ == "__main__":
     main()