Spaces:

M17idd
/

army

Running

M17idd commited on 18 days ago

Commit

ddb39b3

1 Parent(s): 21d2be9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -551,7 +551,7 @@ from transformers import pipeline
 def summarize_text(text):
     summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
-    summary = summarizer(text, max_length=150, min_length=50, do_sample=False)
     return summary[0]['summary_text']
 def find_best_answer(query, top_files, doc_texts):

 def summarize_text(text):
     summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+    summary = summarizer(text, max_length=1024, min_length=512, do_sample=False)
     return summary[0]['summary_text']
 def find_best_answer(query, top_files, doc_texts):