Update app.py

app.py

@@ -29,7 +29,7 @@ def extract_text_from_api(query):
         
 @st.cache_data(ttl=3600) 
 def get_summary(text):
-    model = genai.GenerativeModel('gemini-1.5-pro', generation_config=generation_config, 
+    model = genai.GenerativeModel('gemini-1.5-flash', generation_config=generation_config, 
                                   safety_settings = {
         'HATE': 'BLOCK_NONE',
         'HARASSMENT': 'BLOCK_NONE',
@@ -71,7 +71,7 @@ with tab1:
             if text: 
                 #st.write(text) 
                 summary = get_summary(text)
-                #st.write(summary.candidates[0].safety_ratings)
+                st.write(summary.candidates[0].safety_ratings)
                 if summary:
                     st.write(summary.text) 
             else: