Spaces:
Sleeping
Sleeping
Update doubao_service.py
Browse files- doubao_service.py +3 -2
doubao_service.py
CHANGED
@@ -3,6 +3,7 @@ import requests
|
|
3 |
import json
|
4 |
from volcenginesdkarkruntime import Ark
|
5 |
from util.config_util import read_config as config
|
|
|
6 |
from util import logger
|
7 |
import volcenginesdkcore
|
8 |
import volcenginesdkark
|
@@ -13,8 +14,8 @@ from util.logger_util import log_decorate
|
|
13 |
class DouBaoService:
|
14 |
|
15 |
def __init__(self, model_name):
|
16 |
-
|
17 |
-
self.conf = config
|
18 |
self.client = self.init_client()
|
19 |
self._complete_args = {}
|
20 |
|
|
|
3 |
import json
|
4 |
from volcenginesdkarkruntime import Ark
|
5 |
from util.config_util import read_config as config
|
6 |
+
from util.config_util import load_json
|
7 |
from util import logger
|
8 |
import volcenginesdkcore
|
9 |
import volcenginesdkark
|
|
|
14 |
class DouBaoService:
|
15 |
|
16 |
def __init__(self, model_name):
|
17 |
+
config = load_json('./conf/config.json')
|
18 |
+
self.conf = config[f"{model_name}ModelInfo"]
|
19 |
self.client = self.init_client()
|
20 |
self._complete_args = {}
|
21 |
|