Update app.py

app.py

@@ -51,9 +51,14 @@ class PresentationGenerator:
         login(self.token)
         self.text_model = None
         self.text_tokenizer = None
+        self.loaded_model = None
 
     def load_text_model(self, model_name):
         """Charge le modèle de génération de texte"""
+        if self.loaded_model == model_name:
+            print(f"Modèle {model_name} déjà chargé.")
+            return
+
         model_id = TEXT_MODELS[model_name]
         model_path = None
 
@@ -87,6 +92,8 @@ class PresentationGenerator:
             )
             print(f"Modèle Transformers {model_id} chargé avec succès!")
 
+        self.loaded_model = model_name
+
     def generate_text(self, prompt, temperature=0.7, max_tokens=4096):
         """Génère le texte de la présentation"""
         start_time = time.time()
@@ -103,7 +110,8 @@ class PresentationGenerator:
             outputs = self.text_model.generate(
                 **inputs,
                 max_new_tokens=max_tokens,
-                temperature=temperature
+                temperature=temperature,
+                do_sample=True
             )
             generated_text = self.text_tokenizer.decode(outputs[0], skip_special_tokens=True)
 
@@ -159,18 +167,18 @@ class PresentationGenerator:
 
         return prs
 
-def generate_skeleton(text, text_model_name, temperature, max_tokens):
+def generate_skeleton(text, text_model_name, temperature, max_tokens, progress=gr.Progress()):
     """Génère le squelette de la présentation"""
     try:
         start_time = time.time()
         generator = PresentationGenerator()
 
         # Chargement du modèle de texte uniquement
-        yield "Chargement du modèle...", None, None
+        progress(0, desc="Chargement du modèle...")
         generator.load_text_model(text_model_name)
 
         # Génération du contenu
-        yield "Génération du contenu de la présentation...", None, None
+        progress(0.5, desc="Génération du contenu de la présentation...")
         full_prompt = PREPROMPT + "\n\n" + text
         generated_content = generator.generate_text(full_prompt, temperature, max_tokens)
 
@@ -274,7 +282,8 @@ with gr.Blocks(theme=gr.themes.Glass()) as demo:
             input_text,
             text_model_choice,
             temperature,
-            max_tokens
+            max_tokens,
+            progress_bar
         ],
         outputs=[
             status_output,
@@ -298,3 +307,8 @@ if __name__ == "__main__":
 
 
 
+
+
+
+
+