Spaces:

Iker
/

Translate-100-languages

Running

Iker commited on Mar 28, 2022

Commit

16e9c22

1 Parent(s): 7f116c9

facebook/m2m100_1.2B

Files changed (1) hide show

app.py CHANGED Viewed

@@ -124,7 +124,7 @@ lang_id = {
 @st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def load_model(
-    pretrained_model: str = "tamedai/m2m100_12B",
     cache_dir: str = "models/",
 ):
     tokenizer = M2M100Tokenizer.from_pretrained(pretrained_model, cache_dir=cache_dir)

 @st.cache(suppress_st_warning=True, allow_output_mutation=True)
 def load_model(
+    pretrained_model: str = "facebook/m2m100_1.2B",
     cache_dir: str = "models/",
 ):
     tokenizer = M2M100Tokenizer.from_pretrained(pretrained_model, cache_dir=cache_dir)