Update src/visual_insight.py

src/visual_insight.py

@@ -4,15 +4,11 @@ import json
 import re
 import os
 
+
+hf_token = os.environ.get("HUGGINGFACE_TOKEN")
 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")
 tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token, use_auth_token=True)
 model = AutoModelForCausalLM.from_pretrained(model_id, token=hf_token)