Update app.py

app.py

@@ -5,9 +5,10 @@ from diffusers import FluxPipeline
 import torch
 from PIL import Image
 import fitz  # PyMuPDF pour la gestion des PDF
+import sentencepiece  # Assurez-vous que sentencepiece est installé
 
 def load_pdf(pdf_path):
-    """Extrait le texte d'un fichier PDF"""
+    """Traite le texte d'un fichier PDF"""
     if pdf_path is None:
         return None
     text = ""
@@ -83,10 +84,10 @@ def generate(prompt, reference_file):
                 return generator.generate_image(prompt, pdf_file=reference_file.name)
             elif file_type in ['png', 'jpg', 'jpeg']:
                 return generator.generate_image(prompt, reference_image=reference_file.name)
-        
+
         # Génération sans référence
         return generator.generate_image(prompt)
-    
+
     except Exception as e:
         print(f"Erreur: {str(e)}")
         return None
@@ -104,4 +105,4 @@ demo = gr.Interface(
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()