Spaces:
Sleeping
Sleeping
Update src/insight.py
Browse files- 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)
|