added cachedir im local app

main.py

@@ -1,5 +1,6 @@
 import io
 import time
+import os
 from typing import List, Literal
 from fastapi import FastAPI
 from pydantic import BaseModel
@@ -18,8 +19,9 @@ class TranslationRequest(BaseModel):
 
 
 def load_model(pretrained_model: str = "facebook/m2m100_1.2B", cache_dir: str = "models/"):
-    tokenizer = M2M100Tokenizer.from_pretrained(pretrained_model, cache_dir=cache_dir)
-    model = M2M100ForConditionalGeneration.from_pretrained(pretrained_model, cache_dir=cache_dir).to(device)
+    model_dir = os.path.join(os.getcwd(), cache_dir)
+    tokenizer = M2M100Tokenizer.from_pretrained(pretrained_model, cache_dir=model_dir)
+    model = M2M100ForConditionalGeneration.from_pretrained(pretrained_model, cache_dir=model_dir).to(device)
     model.eval()
     return tokenizer, model