Update app.py

app.py

@@ -1,11 +1,3 @@
-# Add this temporarily at the top of your app to list available models
-import google.generativeai as genai
-genai.configure(api_key=GOOGLE_API_KEY)
-
-for m in genai.list_models():
-    if "generateContent" in m.supported_generation_methods:
-        st.write(m.name)
-
 import os
 import gc
 import torch
@@ -394,7 +386,7 @@ with st.sidebar:
     st.markdown("---")
     st.markdown("### Pipeline Steps")
     st.markdown("""
-**1. Gemini 1.5 Flash** (API)
+**1. Gemini 2.0 Flash** (API)
 Generate 5 captions
 
 **2. BLIP ITM** (Local)
@@ -417,7 +409,7 @@ Caption fusion
     """)
     st.markdown("---")
     st.markdown("**Local:** BLIP ITM, DINO, Qwen2.5")
-    st.markdown("**API:** Gemini 1.5 Flash, Jina")
+    st.markdown("**API:** Gemini 2.0 Flash, Jina")
 
 # ============================================================================
 # MAIN UI
@@ -452,7 +444,7 @@ if uploaded_file is not None:
             progress = st.progress(0)
             status   = st.empty()
 
-            status.info("Step 1/7: Generating captions with Gemini 1.5 Flash...")
+            status.info("Step 1/7: Generating captions with Gemini 2.0 Flash...")
             captions = generate_captions_gemini(input_image)
             progress.progress(14)