kevinpro commited on
Commit
2e7de2e
·
verified ·
1 Parent(s): 61a28dc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -7,7 +7,7 @@ import spaces
7
  import gradio as gr
8
  from sacremoses import MosesPunctNormalizer
9
  from stopes.pipelines.monolingual.utils.sentence_split import get_split_algo
10
- from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
11
  from flores import code_mapping
12
  import platform
13
  import torch
@@ -23,7 +23,7 @@ target_languages = flores_codes # 简化列表
23
  # 假设openai_client已定义,例如:
24
 
25
  device = "cpu" if platform.system() == "Darwin" else "cuda"
26
- MODEL_NAME = "ByteDance-Seed/Seed-X-PPO-7B"
27
 
28
  code_mapping = dict(sorted(code_mapping.items(), key=lambda item: item[0]))
29
  flores_codes = list(code_mapping.keys())
 
7
  import gradio as gr
8
  from sacremoses import MosesPunctNormalizer
9
  from stopes.pipelines.monolingual.utils.sentence_split import get_split_algo
10
+ from transformers import AutoTokenizer, AutoModel
11
  from flores import code_mapping
12
  import platform
13
  import torch
 
23
  # 假设openai_client已定义,例如:
24
 
25
  device = "cpu" if platform.system() == "Darwin" else "cuda"
26
+ MODEL_NAME = "Qwen/Qwen3-0.6B"
27
 
28
  code_mapping = dict(sorted(code_mapping.items(), key=lambda item: item[0]))
29
  flores_codes = list(code_mapping.keys())