Update app.py

app.py

@@ -11,14 +11,28 @@ model_list = [
     # Ajoutez d'autres modèles ici
 ]
 
+
+
+
+model = None
+tokenizer = None
+
 def load_model(model_name):
     """Charge le modèle et le 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
+    global model, tokenizer
+    model = AutoModelForCausalLM.from_pretrained(model_name, trust_remote_code=True)
+    tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+
+
+
+#def load_model(model_name):
+#    """Charge le modèle et le 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"""