daihui.zhang
commited on
Commit
·
df55474
1
Parent(s):
1f45d99
add logging format
Browse files- transcribe/translator.py +1 -1
transcribe/translator.py
CHANGED
|
@@ -28,7 +28,7 @@ class QwenTranslator:
|
|
| 28 |
message = self.to_message(prompt, src_lang, dst_lang)
|
| 29 |
start_time = time.monotonic()
|
| 30 |
output = self.llm.create_chat_completion(messages=message, temperature=0.9)
|
| 31 |
-
|
| 32 |
return output['choices'][0]['message']['content']
|
| 33 |
|
| 34 |
def __call__(self, prompt,*args, **kwargs):
|
|
|
|
| 28 |
message = self.to_message(prompt, src_lang, dst_lang)
|
| 29 |
start_time = time.monotonic()
|
| 30 |
output = self.llm.create_chat_completion(messages=message, temperature=0.9)
|
| 31 |
+
|
| 32 |
return output['choices'][0]['message']['content']
|
| 33 |
|
| 34 |
def __call__(self, prompt,*args, **kwargs):
|