Update src/plan.py

src/plan.py

@@ -4,15 +4,10 @@ import json
 import re
 import os
 
-model_id = "google/gemma-3n-E4B-it"
-
-# Set Hugging Face cache directory
-HF_CACHE_DIR = "./hf_cache"
-os.environ["HF_HOME"] = HF_CACHE_DIR
-os.environ["TRANSFORMERS_CACHE"] = HF_CACHE_DIR
-os.makedirs(HF_CACHE_DIR, exist_ok=True)
 
 hf_token = os.environ.get("HUGGINGFACE_TOKEN")
+model_id = "google/gemma-3n-E4B-it"
+
 tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token, use_auth_token=True)
 model = AutoModelForCausalLM.from_pretrained(model_id, token=hf_token)