Osnly commited on
Commit
1ba4530
·
verified ·
1 Parent(s): 29d5a95

Update src/insight.py

Browse files
Files changed (1) hide show
  1. src/insight.py +0 -5
src/insight.py CHANGED
@@ -5,11 +5,6 @@ import os
5
 
6
  model_id = "google/gemma-3n-E4B-it"
7
 
8
- # Set Hugging Face cache directory
9
- HF_CACHE_DIR = "./hf_cache"
10
- os.environ["HF_HOME"] = HF_CACHE_DIR
11
- os.environ["TRANSFORMERS_CACHE"] = HF_CACHE_DIR
12
- os.makedirs(HF_CACHE_DIR, exist_ok=True)
13
 
14
  hf_token = os.environ.get("HUGGINGFACE_TOKEN")
15
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token, use_auth_token=True)
 
5
 
6
  model_id = "google/gemma-3n-E4B-it"
7
 
 
 
 
 
 
8
 
9
  hf_token = os.environ.get("HUGGINGFACE_TOKEN")
10
  tokenizer = AutoTokenizer.from_pretrained(model_id, token=hf_token, use_auth_token=True)