r

app.py

@@ -1,7 +1,12 @@
 from pathlib import Path
 
 import gradio as gr
-from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
+from transformers import (
+    M2M100ForConditionalGeneration,
+    M2M100Tokenizer,
+    TranslationPipeline,
+    pipeline,
+)
 
 lang_to_code = {
     "Akrikaans": "af",
@@ -135,14 +140,11 @@ lang_names = list(lang_to_code.keys())
 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(
-    pretrained_model_name_or_path="seyoungsong/flores101_mm100_175M",
-    cache_dir=str(cache_dir),
-)
-model = M2M100ForConditionalGeneration.from_pretrained(
-    pretrained_model_name_or_path="seyoungsong/flores101_mm100_175M",
-    cache_dir=str(cache_dir),
+pipe: TranslationPipeline = pipeline(
+    task="translation", model="seyoungsong/flores101_mm100_175M"
 )
+tokenizer: M2M100Tokenizer = pipe.tokenizer
+model: M2M100ForConditionalGeneration = pipe.model
 
 # fix tokenizer
 tokenizer.lang_token_to_id = {