Update app.py

app.py

@@ -13,9 +13,12 @@ model_list = [
 
 def load_model(model_name):
     """Charge le modèle et le tokenizer"""
-    tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
-    model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
-    return model, tokenizer
+    if model_name is not None:
+        tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+        model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
+        return model, tokenizer
+    else:
+        return None, None
 
 def generate_text(model, tokenizer, input_text, max_length, temperature):
     """Génère du texte en utilisant le modèle"""
@@ -25,23 +28,29 @@ def generate_text(model, tokenizer, input_text, max_length, temperature):
 
 def main(model_name, input_text, max_length, temperature):
     """Fonction principale pour générer le texte"""
-    model, tokenizer = load_model(model_name)
-    generated_text = generate_text(model, tokenizer, input_text, max_length, temperature)
-    return generated_text
+    if model_name is not None:
+        model, tokenizer = load_model(model_name)
+        generated_text = generate_text(model, tokenizer, input_text, max_length, temperature)
+        return generated_text
+    else:
+        return "Veuillez sélectionner un modèle"
 
 demo = gr.Blocks()
 
 with demo:
-    gr.Markdown("# Try It")
+    gr.Markdown("# Modèle de Langage")
     
     with gr.Row():
         model_select = gr.Dropdown(model_list, label="Sélectionner un modèle")
+    with gr.Row():
         load_button = gr.Button("Charger le modèle")
         
     with gr.Row():
         input_text = gr.Textbox(label="Texte d'entrée")
+    with gr.Row():
         max_length_slider = gr.Slider(50, 500, label="Longueur maximale", value=200)
         temperature_slider = gr.Slider(0.1, 1.0, label="Température", value=0.7)
+    with gr.Row():
         submit_button = gr.Button("Soumettre")
         
     output_text = gr.Textbox(label="Texte généré")