r

app.py

@@ -137,14 +137,15 @@ 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}")
-pipe: TranslationPipeline = pipeline(
-    task="translation", model="seyoungsong/flores101_mm100_175M"
+model_path = Path("./model_files").resolve()
+print(f"model_path: {model_path}")
+tokenizer: M2M100Tokenizer = M2M100Tokenizer.from_pretrained(
+    pretrained_model_name_or_path=str(model_path), local_files_only=True
 )
-tokenizer: M2M100Tokenizer = pipe.tokenizer
-model: M2M100ForConditionalGeneration = pipe.model
+model = M2M100ForConditionalGeneration.from_pretrained(
+    pretrained_model_name_or_path=str(model_path), local_files_only=True
+)
+
 
 # fix tokenizer
 tokenizer.lang_token_to_id = {