MisterAI commited on
Commit
2adc88e
·
verified ·
1 Parent(s): 7f45bb9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +18 -4
app.py CHANGED
@@ -48,16 +48,30 @@ def main(input_text, max_length, temperature, model_name):
48
  model = None
49
  tokenizer = None
50
 
 
 
 
 
 
 
 
 
 
 
51
  # Fonction pour charger le modèle lors du clic sur le bouton "Charger Le Modèle"
52
  def load_model_button_clicked(model_name):
53
  """Charge le modèle et le tokenizer lors du clic sur le bouton"""
54
- global model, tokenizer
55
  try:
56
  model, tokenizer = load_model(model_name)
57
- return f"Modèle {model_name} chargé avec succès"
 
58
  except Exception as e:
59
- return f"Erreur lors du chargement du modèle {model_name}: {e}"
60
-
 
 
 
61
  #Fonction Pour Conserver un Historique
62
  def append_history(history, message):
63
  return f"{history}\n\n***\n{message}" if history else message
 
48
  model = None
49
  tokenizer = None
50
 
51
+ # Fonction pour charger le modèle lors du clic sur le bouton "Charger Le Modèle"
52
+ #def load_model_button_clicked(model_name):
53
+ # """Charge le modèle et le tokenizer lors du clic sur le bouton"""
54
+ # global model, tokenizer, history
55
+ # try:
56
+ # model, tokenizer = load_model(model_name)
57
+ # return f"Modèle {model_name} chargé avec succès"
58
+ # except Exception as e:
59
+ # return f"Erreur lors du chargement du modèle {model_name}: {e}"
60
+
61
  # Fonction pour charger le modèle lors du clic sur le bouton "Charger Le Modèle"
62
  def load_model_button_clicked(model_name):
63
  """Charge le modèle et le tokenizer lors du clic sur le bouton"""
64
+ global model, tokenizer, history # Ajoutez history ici pour l'utiliser
65
  try:
66
  model, tokenizer = load_model(model_name)
67
+ history = append_history(history, f"Modèle {model_name} chargé avec succès") # Mettre à jour l'historique
68
+ return history # Renvoie l'historique mis à jour
69
  except Exception as e:
70
+ history = append_history(history, f"Erreur lors du chargement du modèle {model_name}: {e}") # Mettre à jour l'historique en cas d'erreur
71
+ return history # Renvoie l'historique mis à jour
72
+
73
+
74
+
75
  #Fonction Pour Conserver un Historique
76
  def append_history(history, message):
77
  return f"{history}\n\n***\n{message}" if history else message