Hugging Face's logo

Spaces:
SumYin
/
M2M-Clone-Test
Runtime error

App Files Community
SumYin commited on
Commit
31e459d
•
1 Parent(s): 534e8dd

change model to facebook/m2m100_418M

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