Update app.py

app.py

@@ -50,35 +50,31 @@ def progress_bar_html(label: str) -> str:
     '''
 
 # TEXT MODEL - Utiliser Napoleon 4B avec configuration modifiée
-model_id = "baconnier/Napoleon_4B_V0.0" 
-
 # Charger la configuration
 config = AutoConfig.from_pretrained(model_id)
 
-# Ajouter tous les attributs manquants
-if not hasattr(config, "vocab_size"):
-    config.vocab_size = 262208  # Valeur correcte pour Gemma 3
-
-if not hasattr(config, "hidden_size"):
-    config.hidden_size = 2560  # Valeur correcte pour Gemma 3 4B
-
-if not hasattr(config, "num_hidden_layers"):
-    config.num_hidden_layers = 34  # Valeur correcte pour Gemma 3 4B
-
-if not hasattr(config, "intermediate_size"):
-    config.intermediate_size = 10240  # Valeur correcte pour Gemma 3 4B
+# Extraire les attributs de text_config vers la configuration principale
+if hasattr(config, "text_config"):
+    for key, value in vars(config.text_config).items():
+        if not hasattr(config, key):
+            setattr(config, key, value)
+else:
+    # Ajouter manuellement les attributs si text_config n'existe pas
+    if not hasattr(config, "vocab_size"):
+        config.vocab_size = 262208
+    if not hasattr(config, "hidden_size"):
+        config.hidden_size = 2560
+    if not hasattr(config, "num_hidden_layers"):
+        config.num_hidden_layers = 34
+    if not hasattr(config, "intermediate_size"):
+        config.intermediate_size = 10240
+    if not hasattr(config, "num_attention_heads"):
+        config.num_attention_heads = 10
+    if not hasattr(config, "sliding_window"):
+        config.sliding_window = 1024
+    if not hasattr(config, "sliding_window_pattern"):
+        config.sliding_window_pattern = 6
 
-if not hasattr(config, "num_attention_heads"):
-    config.num_attention_heads = 10  # Valeur correcte pour Gemma 3 4B
-    
-if not hasattr(config, "sliding_window"):
-    config.sliding_window = 1024  # Valeur correcte pour Gemma 3
-    
-if not hasattr(config, "sliding_window_pattern"):
-    config.sliding_window_pattern = 6  # Valeur basée sur votre config Napoleon
-    
-if not hasattr(config, "rope_scaling"):
-    config.rope_scaling = {"factor": 8.0, "rope_type": "linear"}  # Valeur correcte pour Gemma 3
 
 
 tokenizer = AutoTokenizer.from_pretrained(model_id)