Update app.py

app.py

@@ -13,7 +13,7 @@ import torch
 from llama_cpp import Llama
 import time
 
-# Configuration des modèles disponibles
+# [Configuration des modèles et PREPROMPT restent identiques]
 TEXT_MODELS = {
     "Utter-Project_EuroLLM-1.7B": "utter-project/EuroLLM-1.7B",
     "Mistral Nemo 2407 (GGUF)": "MisterAI/Bartowski_MistralAI_Mistral-Nemo-Instruct-2407-IQ4_XS.gguf",
@@ -111,18 +111,24 @@ class PresentationGenerator:
             )
             return response['choices'][0]['text']
         else:
-            inputs = tokenizer.apply_chat_template(
-                [{"role": "user", "content": prompt}],
+            # Modification ici pour gérer les modèles sans chat template
+            inputs = tokenizer(
+                prompt,
                 return_tensors="pt",
-                return_dict=True
-            )
+                truncation=True,
+                max_length=4096
+            ).to(model.device)
+            
             outputs = model.generate(
                 **inputs,
                 max_new_tokens=max_tokens,
-                temperature=temperature
+                temperature=temperature,
+                do_sample=True,
+                pad_token_id=tokenizer.eos_token_id
             )
             return tokenizer.decode(outputs[0], skip_special_tokens=True)
 
+    # [Le reste de la classe PresentationGenerator reste identique]
     def parse_presentation_content(self, content):
         """Parse le contenu généré en sections pour les diapositives"""
         slides = []
@@ -170,6 +176,7 @@ class PresentationGenerator:
         
         return prs
 
+# [Les fonctions generate_skeleton et create_presentation_file restent identiques]
 def generate_skeleton(text, text_model_name, temperature, max_tokens):
     """Génère le squelette de la présentation"""
     try:
@@ -212,7 +219,7 @@ def create_presentation_file(generated_content):
         print(f"Erreur lors de la création du fichier: {str(e)}")
         return None
 
-# Interface Gradio
+# [L'interface Gradio reste identique]
 with gr.Blocks(theme=gr.themes.Glass()) as demo:
     gr.Markdown(
         """
@@ -295,5 +302,3 @@ with gr.Blocks(theme=gr.themes.Glass()) as demo:
 if __name__ == "__main__":
     demo.launch()
 
-
-