Update app.py
Browse files
app.py
CHANGED
@@ -100,7 +100,8 @@ LANGUAGES = {
|
|
100 |
}
|
101 |
|
102 |
from huggingface_hub import snapshot_download
|
103 |
-
from vllm import LLM
|
|
|
104 |
|
105 |
save_dir = "./model_weights"
|
106 |
repo_id = "ByteDance-Seed/Seed-X-PPO-7B"
|
@@ -128,7 +129,6 @@ decoding_params = BeamSearchParams(beam_width=4,
|
|
128 |
def translate_text(text, source_lang, target_lang):
|
129 |
if not text.strip():
|
130 |
return "请输入要翻译的文本"
|
131 |
-
|
132 |
try:
|
133 |
response = model.generate(messages, decoding_params)
|
134 |
response = [res.outputs[0].text.strip() for res in results]
|
|
|
100 |
}
|
101 |
|
102 |
from huggingface_hub import snapshot_download
|
103 |
+
from vllm import LLM
|
104 |
+
from vllm.sampling_params import BeamSearchParams
|
105 |
|
106 |
save_dir = "./model_weights"
|
107 |
repo_id = "ByteDance-Seed/Seed-X-PPO-7B"
|
|
|
129 |
def translate_text(text, source_lang, target_lang):
|
130 |
if not text.strip():
|
131 |
return "请输入要翻译的文本"
|
|
|
132 |
try:
|
133 |
response = model.generate(messages, decoding_params)
|
134 |
response = [res.outputs[0].text.strip() for res in results]
|