Update app.py

app.py

@@ -19,23 +19,23 @@ device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 # == Model configurations ==
 MODELS = {
-    "Egret XLarge": {
+    "Docling Layout Egret XLarge": {
         "path": "ds4sd/docling-layout-egret-xlarge",
         "model_class": DFineForObjectDetection
     },
-    "Egret Large": {
+    "Docling Layout Egret Large": {
         "path": "ds4sd/docling-layout-egret-large",
         "model_class": DFineForObjectDetection
     },
-    "Egret Medium": {
+    "Docling Layout Egret Medium": {
         "path": "ds4sd/docling-layout-egret-medium", 
         "model_class": DFineForObjectDetection
     },
-    "Heron 101": {
+    "Docling Layout Heron 101": {
         "path": "ds4sd/docling-layout-heron-101",
         "model_class": RTDetrV2ForObjectDetection
     },
-    "Heron": {
+    "Docling Layout Heron": {
         "path": "ds4sd/docling-layout-heron",
         "model_class": RTDetrV2ForObjectDetection
     }
@@ -332,7 +332,7 @@ if __name__ == "__main__":
         gr.HTML("""
         <div style='text-align: center; padding: 20px; background: linear-gradient(135deg, #667eea 0%, #764ba2 100%); color: white; border-radius: 12px; margin-bottom: 20px;'>
             <h1 style='margin: 0; font-size: 2.5em;'>🔍 Document Layout Analysis</h1>
-            <p style='margin: 8px 0 0 0; font-size: 1.1em; opacity: 0.9;'>Advanced document structure detection</p>
+            <p style='margin: 8px 0 0 0; font-size: 1.1em; opacity: 0.9;'>Advanced document structure detection with Docling models </p>
         </div>
         """)
         
@@ -355,7 +355,7 @@ if __name__ == "__main__":
                     # 2. Model Selection (second, without buttons)
                     model_dropdown = gr.Dropdown(
                         choices=list(MODELS.keys()),
-                        value="Egret XLarge",
+                        value="Docling Layout Egret XLarge",
                         label="AI Model",
                         info="Model will be loaded automatically",
                         interactive=True