fixe

app.py

@@ -53,7 +53,7 @@ logger =logging.getLogger(__name__)
 # Définir le modèle de requête
 class PredictionRequest(BaseModel):
     text: str = None  # Texte personnalisé ajouté par l'utilisateur
-    max_length: int = 2000  # Limite la longueur maximale du texte généré
+    # max_length: int = 2000  # Limite la longueur maximale du texte généré
 
 @app.post("/predict/")
 async def predict(request: PredictionRequest):
@@ -65,7 +65,7 @@ async def predict(request: PredictionRequest):
 
     # Générer le texte à partir du prompt
     inputs = tokenizer(prompt, return_tensors="pt").input_ids.to(model.device)
-    outputs = model.generate(inputs, max_length=request.max_length, do_sample=True)
+    outputs = model.generate(inputs, max_length=3000, do_sample=True)
     generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
     
     return {"generated_text": generated_text}