Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,6 +21,10 @@ from pydantic import BaseModel
|
|
| 21 |
# -------------------------------
|
| 22 |
# Utils
|
| 23 |
# -------------------------------
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
load_dotenv()
|
| 25 |
GROQ_API_KEY = None
|
| 26 |
hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
|
|
|
|
| 21 |
# -------------------------------
|
| 22 |
# Utils
|
| 23 |
# -------------------------------
|
| 24 |
+
os.environ["HF_HOME"] = "/tmp/hf_cache"
|
| 25 |
+
os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
|
| 26 |
+
os.environ["XDG_CACHE_HOME"] = "/tmp/hf_cache"
|
| 27 |
+
|
| 28 |
load_dotenv()
|
| 29 |
GROQ_API_KEY = None
|
| 30 |
hf_token = os.getenv("HUGGINGFACEHUB_API_TOKEN")
|