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)
|