Spaces:
Running
Running
Use m2m100_12B
Browse files
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 = "
|
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)
|