Update app.py

app.py

@@ -6,7 +6,7 @@ Features:
   - 🖼️ Image Analysis (with Face Detection, Signature Extraction, Annotated Highlights)
   - 📚 Batch Processing
   - 💬 Chat with Attachments
-All 10 Qwen VL models available across all tabs.
+All 5 Chhagan VL models available across all tabs.
 """
 
 import os
@@ -120,17 +120,9 @@ device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 print(f"🚀 Using device: {device}")
 
 # ──────────────────────────────────────────────────────────────
-# ALL MODELS
+# ALL MODELS  ←── CHANGE 1: Only Chhagan custom models remain
 # ──────────────────────────────────────────────────────────────
 ALL_MODELS = [
-    # ── Qwen3-VL Official ──
-    "Qwen/Qwen3-VL-2B-Instruct",
-    "Qwen/Qwen3-VL-4B-Instruct",
-    "Qwen/Qwen3-VL-8B-Instruct",
-    # ── Qwen2.5-VL Official ──
-    "Qwen/Qwen2.5-VL-7B-Instruct",
-    "Qwen/Qwen2.5-VL-3B-Instruct",
-    # ── Custom Models ──
     "Chhagan005/Chhagan_ML-VL-OCR-v1",               # Qwen2.5-VL-3B base
     "Chhagan005/Chhagan-DocVL-Qwen3",                 # Qwen3-VL-2B LoRA adapter
     "Chhagan005/CSM-DocExtract-VL",                   # Qwen3-VL-8B BNB INT4
@@ -240,9 +232,9 @@ def load_model(model_id: str):
     return processor, model
 
 # ──────────────────────────────────────────────────────────────
-# Pre-load default model
+# Pre-load default model  ←── CHANGE 2: New default model
 # ──────────────────────────────────────────────────────────────
-DEFAULT_MODEL = "Qwen/Qwen3-VL-8B-Instruct"
+DEFAULT_MODEL = "Chhagan005/CSM-DocExtract-VL-HF"
 print(f"⏳ Pre-loading default model: {DEFAULT_MODEL}")
 load_model(DEFAULT_MODEL)
 print(f"✅ Default model ready!")
@@ -596,7 +588,8 @@ with gr.Blocks(title="Chhagan's Multi-Model Studio") as demo:
 
     gr.Markdown("# 🧠 Chhagan's  Multi-Model Studio", elem_id="app-title")
     gr.Markdown(
-        "_Qwen3-VL + Qwen2.5-VL Official + Custom CSM/Chhagan VL models — "
+        # ←── CHANGE 3: Updated subtitle (removed Qwen official mention)
+        "_Custom CSM/Chhagan VL models — "
         "Document Scanning • Image Analysis (Face + Signature) • Batch Processing • Chat_",
         elem_id="app-subtitle",
     )
@@ -604,7 +597,8 @@ with gr.Blocks(title="Chhagan's Multi-Model Studio") as demo:
     with gr.Row():
         model_choice = gr.Dropdown(choices=ALL_MODELS, value=DEFAULT_MODEL,
             label="🤖 Select Model",
-            info="3 Qwen3-VL + 2 Qwen2.5-VL official + 5 custom CSM/Chhagan models",
+            # ←── CHANGE 4: Updated info text
+            info="5 custom CSM/Chhagan models",
             scale=3, elem_classes="model-selector")
 
     with gr.Accordion("⚙️ Advanced Generation Parameters", open=False):
@@ -712,8 +706,8 @@ with gr.Blocks(title="Chhagan's Multi-Model Studio") as demo:
             clear_btn.click(clear_fn, outputs=[chat_msg, chatbot],                            queue=False)
 
     gr.Markdown(
-        "---\n**🧠 Chhagan's  Multi-Model Studio** • 10 Models\n\n"
-        "Qwen3-VL (2B/4B/8B) • Qwen2.5-VL (3B/7B) • "
+        # ←── CHANGE 5: Updated footer (removed Qwen official models, 10→5)
+        "---\n**🧠 Chhagan's  Multi-Model Studio** • 5 Models\n\n"
         "CSM-DocExtract-VL (BNB-INT4) • CSM-DocExtract-VL-HF • "
         "CSM-DocExtract-VL-Q4KM-merged-fp16 • Chhagan_ML-VL-OCR-v1 • Chhagan-DocVL-Qwen3\n\n"
         "_Built with ❤️ using Gradio_"