Update app.py

app.py

@@ -31,6 +31,10 @@ def pdfopen(filepath : str) -> str:
     
     return text.strip()
 
+def clean_short(ans):
+    words = ans.split()
+    return " ".join(words[:3])
+
 
 def summarizer():
     
@@ -46,7 +50,7 @@ def anq():
     
     tokenizer = AutoTokenizer.from_pretrained(qnap, use_fast=False)
     model = AutoModelForQuestionAnswering.from_pretrained(qnap)
-    k = pipeline("question-answering", model=model, tokenizer=tokenizer)
+    k = pipeline("question-answering", model=model, tokenizer=tokenizer,max_answer_len=5)
     
     return k
 
@@ -164,10 +168,14 @@ def quesans(py : qna):
         raise HTTPException(status_code=400, detail="No PDF text found. Upload PDF first.")
     
     g = anq()
+
+    forced_question = py.question + " (Give answer in 1 to 3 words only.)"
+
+    result = g (question= forced_question, context= txt2)
+
+    cleaned = clean_short(ans = result["answer"])
     
-    result = g (question= py.question, context= txt2)
-    
-    return {"answer" : result["answer"]}
+    return {"answer" : cleaned}
     
     
 @app.post("/clausedetection")