From dbbc8d25ac35c609fa02472686ed98acf714ea8d Mon Sep 17 00:00:00 2001
From: kwaroran <kwaworan@gmail.com>
Date: Sat, 5 Aug 2023 23:13:32 +0900
Subject: [PATCH] [feat] add transformer types

---
 src/ts/process/embedding/transformers.ts | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/ts/process/embedding/transformers.ts b/src/ts/process/embedding/transformers.ts
index bb2170d2..d1bd7999 100644
--- a/src/ts/process/embedding/transformers.ts
+++ b/src/ts/process/embedding/transformers.ts
@@ -1,6 +1,4 @@
 import type { Pipeline, PretrainedOptions } from '@xenova/transformers';
-import { DataBase } from 'src/ts/storage/database';
-import { get } from 'svelte/store';
 
 let pipeline: (task: string, model?: string, { quantized, progress_callback, config, cache_dir, local_files_only, revision, }?: PretrainedOptions) => Promise<Pipeline> = null
 
@@ -37,7 +35,7 @@ export const runTransformers = async (baseText:string, model:string,bodyTemplate
     await loadTransformer()
     let text = baseText
     let generator = await pipeline('text-generation', model);
-    let output = await generator(text, bodyTemplate);
+    let output:{generated_text:string}[] = await generator(text);
     return output
 }