Update app.py

app.py

@@ -28,6 +28,23 @@ models_and_languages = {
     "croissantllm/CroissantLLMBase": ["en", "fr"]
 }
 
+# Paramètres recommandés pour chaque modèle
+model_parameters = {
+    "meta-llama/Llama-2-13b-hf": {"temperature": 0.8, "top_p": 0.9, "top_k": 40},
+    "meta-llama/Llama-2-7b-hf": {"temperature": 0.8, "top_p": 0.9, "top_k": 40},
+    "meta-llama/Llama-2-70b-hf": {"temperature": 0.8, "top_p": 0.9, "top_k": 40},
+    "meta-llama/Meta-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},
+    "mistralai/Mistral-7B-v0.1": {"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-2-2b": {"temperature": 0.7, "top_p": 0.95, "top_k": 40},
+    "google/gemma-2-9b": {"temperature": 0.7, "top_p": 0.95, "top_k": 40},
+    "google/gemma-2-27b": {"temperature": 0.7, "top_p": 0.95, "top_k": 40},
+    "croissantllm/CroissantLLMBase": {"temperature": 0.8, "top_p": 0.92, "top_k": 50}
+}
+
 # Variables globales
 model = None
 tokenizer = None
@@ -60,9 +77,18 @@ def load_model(model_name, progress=gr.Progress()):
         
         progress(1.0, desc="Modèle chargé")
         available_languages = models_and_languages[model_name]
-        return f"Modèle {model_name} chargé avec succès. Langues disponibles : {', '.join(available_languages)}", gr.Dropdown(choices=available_languages, value=available_languages[0], visible=True, interactive=True)
+        
+        # Mise à jour des sliders avec les valeurs recommandées
+        params = model_parameters[model_name]
+        return (
+            f"Modèle {model_name} chargé avec succès. Langues disponibles : {', '.join(available_languages)}",
+            gr.Dropdown(choices=available_languages, value=available_languages[0], visible=True, interactive=True),
+            params["temperature"],
+            params["top_p"],
+            params["top_k"]
+        )
     except Exception as e:
-        return f"Erreur lors du chargement du modèle : {str(e)}", gr.Dropdown(visible=False)
+        return f"Erreur lors du chargement du modèle : {str(e)}", gr.Dropdown(visible=False), None, None, None
 
 def set_language(lang):
     global selected_language
@@ -177,7 +203,7 @@ def reset():
     return "", 1.0, 1.0, 50, None, None, None, None, gr.Dropdown(visible=False), ""
 
 with gr.Blocks() as demo:
-    gr.Markdown("# Analyse et génération de texte")
+    gr.Markdown("# LLM&BIAS")
     
     with gr.Accordion("Sélection du modèle"):
         model_dropdown = gr.Dropdown(choices=list(models_and_languages.keys()), label="Choisissez un modèle")
@@ -205,7 +231,9 @@ with gr.Blocks() as demo:
     
     reset_button = gr.Button("Réinitialiser")
     
-    load_button.click(load_model, inputs=[model_dropdown], outputs=[load_output, language_dropdown])
+    load_button.click(load_model, 
+                      inputs=[model_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],