Update app.py

app.py

@@ -11,7 +11,7 @@ import torch
 from pdf2image import convert_from_path
 from PIL import Image
 from torch.utils.data import DataLoader
-from transformers import AutoProcessor, pipeline
+from transformers import AutoProcessor
 
 sys.path.append(os.path.abspath(os.path.join(os.path.dirname(__file__), './colpali-main')))
 
@@ -41,14 +41,6 @@ mock_image = Image.new("RGB", (448, 448), (255, 255, 255))
 ds = []
 images = []
 
-# Initialiser le pipeline de génération de texte
-pipe = pipeline("text2text-generation", model="google/flan-t5-small")
-
-@app.get("/generate")
-def generate(text: str):
-    output = pipe(text)
-    return {"output": output[0]["generated_text"]}
-
 # Rediriger la racine vers /docs
 @app.get("/")
 def read_root():