From fb941997546186ae182cfc8eec8c24937eca561f Mon Sep 17 00:00:00 2001
From: superobk <superobk@hotmail.com>
Date: Thu, 28 Mar 2024 17:31:42 +0800
Subject: [PATCH] refactor: relocate asr model

---
 src/{asr => blackbox}/asr.py     | 6 +++---
 src/blackbox/blackbox_factory.py | 2 +-
 src/blackbox/tesou.py            | 1 -
 3 files changed, 4 insertions(+), 5 deletions(-)
 rename src/{asr => blackbox}/asr.py (90%)

diff --git a/src/asr/asr.py b/src/blackbox/asr.py
similarity index 90%
rename from src/asr/asr.py
rename to src/blackbox/asr.py
index fa879f8..7ba88fb 100644
--- a/src/asr/asr.py
+++ b/src/blackbox/asr.py
@@ -4,9 +4,9 @@ from typing import Any, Coroutine
 from fastapi import Request, Response, status
 from fastapi.responses import JSONResponse
 
-from .rapid_paraformer.utils import read_yaml
-from .rapid_paraformer import RapidParaformer
-from ..blackbox.blackbox import Blackbox
+from ..asr.rapid_paraformer.utils import read_yaml
+from ..asr.rapid_paraformer import RapidParaformer
+from .blackbox import Blackbox
 
 class ASR(Blackbox):
 
diff --git a/src/blackbox/blackbox_factory.py b/src/blackbox/blackbox_factory.py
index d2828ca..973b8a3 100644
--- a/src/blackbox/blackbox_factory.py
+++ b/src/blackbox/blackbox_factory.py
@@ -1,7 +1,7 @@
 from .sum import SUM
 from .sentiment import Sentiment
 from .tts import TTS
-from ..asr.asr import ASR
+from .asr import ASR
 from .audio_to_text import AudioToText
 from .blackbox import Blackbox
 from .calculator import Calculator
diff --git a/src/blackbox/tesou.py b/src/blackbox/tesou.py
index d07d898..6b95d3f 100755
--- a/src/blackbox/tesou.py
+++ b/src/blackbox/tesou.py
@@ -23,7 +23,6 @@ class Tesou(Blackbox):
             "user_id": id,
             "prompt": prompt,
         }
-        print(message)
         response = requests.post(url, json=message)
         return response.json()