Using try_to_load_from_cache() with repo to store model

app/services/ai_service.py

@@ -11,15 +11,15 @@ GPT2_MODEL_NAME = "distilgpt2"
 # os.makedirs(CACHE_DIR, exist_ok=True)
 
 
-for lock_file in glob.glob(os.path.join(try_to_load_from_cache(), "*.lock")):
+for lock_file in glob.glob(os.path.join(try_to_load_from_cache('funny-meme-generator', 'cache_dir'), "*.lock")):
     try:
         os.remove(lock_file)
         print(f"Removed lock file: {lock_file}")
     except Exception as e:
         print(f"Could not remove {lock_file}: {e}")
 
-gpt2_tokenizer = AutoTokenizer.from_pretrained(GPT2_MODEL_NAME, cache_dir=try_to_load_from_cache())
-gpt2_model = AutoModelForCausalLM.from_pretrained(GPT2_MODEL_NAME, cache_dir=try_to_load_from_cache())
+gpt2_tokenizer = AutoTokenizer.from_pretrained(GPT2_MODEL_NAME, cache_dir=try_to_load_from_cache('funny-meme-generator', 'cache_dir'))
+gpt2_model = AutoModelForCausalLM.from_pretrained(GPT2_MODEL_NAME, cache_dir=try_to_load_from_cache('funny-meme-generator', 'cache_dir'))
 
 # tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
 # model = AutoModelWithLMHead.from_pretrained(MODEL_NAME)