Iker commited on
Commit
7f116c9
1 Parent(s): 3d3c0dc

Use m2m100_12B

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -124,7 +124,7 @@ lang_id = {
124
 
125
  @st.cache(suppress_st_warning=True, allow_output_mutation=True)
126
  def load_model(
127
- pretrained_model: str = "facebook/m2m100_418M",
128
  cache_dir: str = "models/",
129
  ):
130
  tokenizer = M2M100Tokenizer.from_pretrained(pretrained_model, cache_dir=cache_dir)
 
124
 
125
  @st.cache(suppress_st_warning=True, allow_output_mutation=True)
126
  def load_model(
127
+ pretrained_model: str = "tamedai/m2m100_12B",
128
  cache_dir: str = "models/",
129
  ):
130
  tokenizer = M2M100Tokenizer.from_pretrained(pretrained_model, cache_dir=cache_dir)