Update app.py
Browse files
app.py
CHANGED
|
@@ -10,6 +10,9 @@ import uvicorn
|
|
| 10 |
import tempfile
|
| 11 |
from time import time
|
| 12 |
from pyngrok import ngrok
|
|
|
|
|
|
|
|
|
|
| 13 |
|
| 14 |
# Add necessary serialization safety
|
| 15 |
torch.serialization.add_safe_globals([DynamicCache])
|
|
|
|
| 10 |
import tempfile
|
| 11 |
from time import time
|
| 12 |
from pyngrok import ngrok
|
| 13 |
+
|
| 14 |
+
os.environ["HF_HOME"] = "/app/hf_cache"
|
| 15 |
+
os.environ["TRANSFORMERS_CACHE"] = "/app/hf_cache"
|
| 16 |
|
| 17 |
# Add necessary serialization safety
|
| 18 |
torch.serialization.add_safe_globals([DynamicCache])
|