r

app.py

@@ -1,3 +1,5 @@
+from pathlib import Path
+
 import gradio as gr
 from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
 
@@ -130,11 +132,16 @@ lang_to_code = {
 lang_names = list(lang_to_code.keys())
 
 # load model
+cache_dir = Path("./cache_dir").resolve()
+cache_dir.mkdir(exist_ok=True, parents=True)
+print(f"cache_dir: {cache_dir}")
 tokenizer: M2M100Tokenizer = M2M100Tokenizer.from_pretrained(
-    "seyoungsong/flores101_mm100_175M"
+    pretrained_model_name_or_path="seyoungsong/flores101_mm100_175M",
+    cache_dir=str(cache_dir),
 )
 model = M2M100ForConditionalGeneration.from_pretrained(
-    "seyoungsong/flores101_mm100_175M"
+    pretrained_model_name_or_path="seyoungsong/flores101_mm100_175M",
+    cache_dir=str(cache_dir),
 )
 
 # fix tokenizer