YuLu0713 commited on
Commit
feda65f
·
verified ·
1 Parent(s): ea4d01d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -100,7 +100,8 @@ LANGUAGES = {
100
  }
101
 
102
  from huggingface_hub import snapshot_download
103
- from vllm import LLM, SamplingParams, BeamSearchParams
 
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]