From 9f57b05801f8cbcbd8caaf98b1b88fb985d4954c Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Wed, 5 Nov 2025 14:54:48 +0800
Subject: [PATCH] bugfix

---
 conf/config.json |  4 ++++
 fw/fw.py         | 11 ++++++++---
 fw/init.py       |  4 +---
 3 files changed, 13 insertions(+), 6 deletions(-)

diff --git a/conf/config.json b/conf/config.json
index b013399..a2754cf 100644
--- a/conf/config.json
+++ b/conf/config.json
@@ -1,5 +1,9 @@
 {   
     "filesroot":"$[workdir]$/files",
+
+	"model_path": "/data/ymq/models/openai/whisper-large-v3",
+	"redis_url": "redis://127.0.0.1:6379",
+	"worker_cnt": 1,
     "logger":{
         "name":"fastwhisper",
         "levelname":"info",
diff --git a/fw/fw.py b/fw/fw.py
index 1e68491..9622eb3 100644
--- a/fw/fw.py
+++ b/fw/fw.py
@@ -1,17 +1,22 @@
 from longtasks.longtasks import LongTasks
 from faster_whisper import WhisperModel
 from appPublic.worker import awaitify
+from appPublic.jsonConfig import getConfig
 from ahserver.filestorage import FileStorage
 
 class MyLongTask(LongTasks):
-	def __init__(self, *args, **kw):
-		super().__init__(self, *args, **kw)
+	def __init__(self):
+		self.config = getConfig()
+		redis_url = self.config.redis_url
+		taskname = 'fastwhisper'
+		worker_cnt = self.config.worker_cnt
+		super().__init__(self, redis_url, taskname, worker_cnt=worker_cnt)
 		self.load_models()
 	
 	def load_models(self):
 		self.models = []
 		for i in range(self.worker_cnt):
-			model = WhisperModel(model_size, device="cuda", compute_type="float16")
+			model = WhisperModel(self.config.model_path, device="cuda", compute_type="float16")
 			self.models.append(model)
 	
 	async def process_task(self, payload:dict, workerid:int=None):
diff --git a/fw/init.py b/fw/init.py
index cc4c35c..fd049ef 100644
--- a/fw/init.py
+++ b/fw/init.py
@@ -1,11 +1,9 @@
 from fastwhisper.fw import MyLongTask
 from ahserver.serverenv import ServerEnv
-from appPublic.jsonConfig import getConfig
 from appPublic.worker import schedule_once
 
 def load_fastwhisper()
-	config = getConfig()
 	env = ServerEnv()
-	engine = MyLongTask(config.redis_url, 'fastwhisper')
+	engine = MyLongTask()
 	schedule_once(1, engine.run)
 	env.fastwhisper = engine