tuandunghcmut commited on
Commit
492390c
·
verified ·
1 Parent(s): 4d0bc11

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -39,7 +39,8 @@ LANGUAGES = {
39
  from transformers import AutoModelForCausalLM, AutoTokenizer
40
 
41
  device = "cuda"
42
- MODEL_NAME = "ByteDance-Seed/Seed-X-PPO-7B"
 
43
  print("Start dowload")
44
  def load_model():
45
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME,torch_dtype="bfloat16").to(device)
@@ -151,9 +152,9 @@ with gr.Blocks(title="Seed-X") as demo:
151
  outputs=target_text
152
  )
153
 
154
- gr.Markdown(
155
- "🌐[Github](https://github.com/ByteDance-Seed/Seed-X-7B)  📄[Report](https://arxiv.org/pdf/2507.13618)  🤗[Model](https://huggingface.co/collections/ByteDance-Seed/seed-x-6878753f2858bc17afa78543)"
156
- )
157
 
158
  # 启动应用
159
  if __name__ == "__main__":
 
39
  from transformers import AutoModelForCausalLM, AutoTokenizer
40
 
41
  device = "cuda"
42
+ # MODEL_NAME = "ByteDance-Seed/Seed-X-PPO-7B"
43
+ MODEL_NAME="Qwen/Qwen3-Next-80B-A3B-Instruct"
44
  print("Start dowload")
45
  def load_model():
46
  model = AutoModelForCausalLM.from_pretrained(MODEL_NAME,torch_dtype="bfloat16").to(device)
 
152
  outputs=target_text
153
  )
154
 
155
+ # gr.Markdown(
156
+ # "🌐[Github](https://github.com/ByteDance-Seed/Seed-X-7B)  📄[Report](https://arxiv.org/pdf/2507.13618)  🤗[Model](https://huggingface.co/collections/ByteDance-Seed/seed-x-6878753f2858bc17afa78543)"
157
+ # )
158
 
159
  # 启动应用
160
  if __name__ == "__main__":