Update app.py

app.py

@@ -12,10 +12,10 @@ import time
 login(token=os.environ["HF_TOKEN"])
 
 # Structure hiérarchique des modèles
-model_hierarchy = {
+models_hierarchy = {
     "meta-llama": {
         "Llama-2": ["7B", "13B", "70B"],
-        "Llama-3": ["8B", "3.2-3B", "3.1-8B"]
+        "Llama-3": ["8B", "3.2B", "3.1B"]
     },
     "mistralai": {
         "Mistral": ["7B-v0.1", "7B-v0.3"],
@@ -30,16 +30,16 @@ model_hierarchy = {
 }
 
 # Langues supportées par modèle
-models_and_languages = {
+models_languages = {
     "meta-llama/Llama-2-7B": ["en"],
     "meta-llama/Llama-2-13B": ["en"],
     "meta-llama/Llama-2-70B": ["en"],
     "meta-llama/Llama-3-8B": ["en"],
-    "meta-llama/Llama-3.2-3B": ["en", "de", "fr", "it", "pt", "hi", "es", "th"],
-    "meta-llama/Llama-3.1-8B": ["en", "de", "fr", "it", "pt", "hi", "es", "th"],
+    "meta-llama/Llama-3-3.2B": ["en", "de", "fr", "it", "pt", "hi", "es", "th"],
+    "meta-llama/Llama-3-3.1B": ["en", "de", "fr", "it", "pt", "hi", "es", "th"],
     "mistralai/Mistral-7B-v0.1": ["en"],
-    "mistralai/Mistral-7B-v0.3": ["en"],
     "mistralai/Mixtral-8x7B-v0.1": ["en", "fr", "it", "de", "es"],
+    "mistralai/Mistral-7B-v0.3": ["en"],
     "google/Gemma-2B": ["en"],
     "google/Gemma-9B": ["en"],
     "google/Gemma-27B": ["en"],
@@ -52,11 +52,11 @@ model_parameters = {
     "meta-llama/Llama-2-13B": {"temperature": 0.8, "top_p": 0.9, "top_k": 40},
     "meta-llama/Llama-2-70B": {"temperature": 0.8, "top_p": 0.9, "top_k": 40},
     "meta-llama/Llama-3-8B": {"temperature": 0.75, "top_p": 0.9, "top_k": 50},
-    "meta-llama/Llama-3.2-3B": {"temperature": 0.75, "top_p": 0.9, "top_k": 50},
-    "meta-llama/Llama-3.1-8B": {"temperature": 0.75, "top_p": 0.9, "top_k": 50},
+    "meta-llama/Llama-3-3.2B": {"temperature": 0.75, "top_p": 0.9, "top_k": 50},
+    "meta-llama/Llama-3-3.1B": {"temperature": 0.75, "top_p": 0.9, "top_k": 50},
     "mistralai/Mistral-7B-v0.1": {"temperature": 0.7, "top_p": 0.9, "top_k": 50},
-    "mistralai/Mistral-7B-v0.3": {"temperature": 0.7, "top_p": 0.9, "top_k": 50},
     "mistralai/Mixtral-8x7B-v0.1": {"temperature": 0.8, "top_p": 0.95, "top_k": 50},
+    "mistralai/Mistral-7B-v0.3": {"temperature": 0.7, "top_p": 0.9, "top_k": 50},
     "google/Gemma-2B": {"temperature": 0.7, "top_p": 0.95, "top_k": 40},
     "google/Gemma-9B": {"temperature": 0.7, "top_p": 0.95, "top_k": 40},
     "google/Gemma-27B": {"temperature": 0.7, "top_p": 0.95, "top_k": 40},
@@ -69,10 +69,10 @@ tokenizer = None
 selected_language = None
 
 def update_model_choices(company):
-    return gr.Dropdown(choices=list(model_hierarchy[company].keys()), value=None)
+    return list(models_hierarchy[company].keys())
 
 def update_variation_choices(company, model_name):
-    return gr.Dropdown(choices=model_hierarchy[company][model_name], value=None)
+    return models_hierarchy[company][model_name]
 
 def load_model(company, model_name, variation, progress=gr.Progress()):
     global model, tokenizer
@@ -101,9 +101,9 @@ def load_model(company, model_name, variation, progress=gr.Progress()):
             tokenizer.pad_token = tokenizer.eos_token
         
         progress(1.0, desc="Modèle chargé")
-        available_languages = models_and_languages[full_model_name]
-        params = model_parameters[full_model_name]
+        available_languages = models_languages[full_model_name]
         
+        params = model_parameters[full_model_name]
         return (
             f"Modèle {full_model_name} chargé avec succès. Langues disponibles : {', '.join(available_languages)}",
             gr.Dropdown(choices=available_languages, value=available_languages[0], visible=True, interactive=True),
@@ -223,20 +223,15 @@ def reset():
     model = None
     tokenizer = None
     selected_language = None
-    return (
-        gr.Dropdown(choices=list(model_hierarchy.keys()), value=None),
-        gr.Dropdown(visible=False),
-        gr.Dropdown(visible=False),
-        "", 1.0, 1.0, 50, None, None, None, None, gr.Dropdown(visible=False), ""
-    )
+    return "", 1.0, 1.0, 50, None, None, None, None, gr.Dropdown(visible=False), ""
 
 with gr.Blocks() as demo:
     gr.Markdown("# LLM&BIAS")
     
     with gr.Accordion("Sélection du modèle"):
-        company_dropdown = gr.Dropdown(choices=list(model_hierarchy.keys()), label="Choisissez une société")
-        model_dropdown = gr.Dropdown(label="Choisissez un modèle", visible=False)
-        variation_dropdown = gr.Dropdown(label="Choisissez une variation", visible=False)
+        company_dropdown = gr.Dropdown(choices=list(models_hierarchy.keys()), label="Choisissez une société")
+        model_dropdown = gr.Dropdown(label="Choisissez un modèle", interactive=False)
+        variation_dropdown = gr.Dropdown(label="Choisissez une variation", interactive=False)
         load_button = gr.Button("Charger le modèle")
         load_output = gr.Textbox(label="Statut du chargement")
         language_dropdown = gr.Dropdown(label="Choisissez une langue", visible=False)
@@ -263,9 +258,10 @@ with gr.Blocks() as demo:
     
     company_dropdown.change(update_model_choices, inputs=[company_dropdown], outputs=[model_dropdown])
     model_dropdown.change(update_variation_choices, inputs=[company_dropdown, model_dropdown], outputs=[variation_dropdown])
+    
     load_button.click(load_model, 
                       inputs=[company_dropdown, model_dropdown, variation_dropdown], 
-                      outputs=[load_output, language_dropdown])
+                      outputs=[load_output, language_dropdown, temperature, top_p, top_k])
     language_dropdown.change(set_language, inputs=[language_dropdown], outputs=[language_output])
     analyze_button.click(analyze_next_token, 
                          inputs=[input_text, temperature, top_p, top_k],