Update app.py

app.py

@@ -1,6 +1,9 @@
-##Modif 138
+#139
 
+
+import os
 import gradio as gr
+from huggingface_hub import hf_hub_download, login
 from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
 from pptx import Presentation
 from pptx.util import Inches, Pt
@@ -80,6 +83,7 @@ class PresentationGenerator:
                 n_batch=512,
                 verbose=False
             )
+            print(f"Modèle GGUF {model_id} chargé avec succès!")
         else:
             # Configuration pour les modèles Transformers standards
             self.text_tokenizer = AutoTokenizer.from_pretrained(model_id, token=self.token)
@@ -89,6 +93,7 @@ class PresentationGenerator:
                 device_map="auto",
                 token=self.token
             )
+            print(f"Modèle Transformers {model_id} chargé avec succès!")
 
     def load_image_model(self, model_name):
         """Charge le modèle de génération d'images"""
@@ -105,6 +110,7 @@ class PresentationGenerator:
                 temperature=temperature,
                 echo=False
             )
+            print("Texte généré par Llama :", response['choices'][0]['text'])
             return response['choices'][0]['text']
         else:
             inputs = self.text_tokenizer.apply_chat_template(
@@ -117,7 +123,9 @@ class PresentationGenerator:
                 max_new_tokens=max_tokens,
                 temperature=temperature
             )
-            return self.text_tokenizer.decode(outputs[0], skip_special_tokens=True)
+            generated_text = self.text_tokenizer.decode(outputs[0], skip_special_tokens=True)
+            print("Texte généré par Transformers :", generated_text)
+            return generated_text
 
     def generate_image(self, prompt, negative_prompt="", num_inference_steps=30):
         """Génère une image pour la diapositive"""