[feat] improve supa/hypa memory

2023-09-08 01:58:28 +09:00
parent 677bd51657
commit e109cd2dc8
6 changed files with 60 additions and 596 deletions
--- a/src/ts/process/embedding/transformers.ts
+++ b/src/ts/process/embedding/transformers.ts
@@ -42,7 +42,7 @@ export const runTransformers = async (baseText:string, model:string,bodyTemplate

 export const runSummarizer = async (text: string) => {
    await loadTransformer()
-    let classifier = await pipeline("summarization", "Xenova/bart-large-cnn")
+    let classifier = await pipeline("summarization", "Xenova/distilbart-cnn-6-6")
    const v:{summary_text:string}[] = await classifier(text)
    return v
 }