Update main.py
Browse files
main.py
CHANGED
@@ -3,7 +3,7 @@ from fastapi import FastAPI, Request
|
|
3 |
from pydantic import BaseModel
|
4 |
from transformers import pipeline
|
5 |
|
6 |
-
# Set cache
|
7 |
os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
|
8 |
|
9 |
app = FastAPI()
|
|
|
3 |
from pydantic import BaseModel
|
4 |
from transformers import pipeline
|
5 |
|
6 |
+
# ✅ Set cache to writable directory
|
7 |
os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
|
8 |
|
9 |
app = FastAPI()
|