Spaces:
Running
Running
updated app.py and docker for proper permissions to data dir
Browse files- Dockerfile +1 -0
- app.py +2 -2
Dockerfile
CHANGED
|
@@ -3,6 +3,7 @@ FROM python:3.12-slim
|
|
| 3 |
RUN apt-get update && apt-get install -y git
|
| 4 |
|
| 5 |
RUN mkdir -p /data/.cache && chmod -R 777 /data/.cache
|
|
|
|
| 6 |
|
| 7 |
WORKDIR /app
|
| 8 |
|
|
|
|
| 3 |
RUN apt-get update && apt-get install -y git
|
| 4 |
|
| 5 |
RUN mkdir -p /data/.cache && chmod -R 777 /data/.cache
|
| 6 |
+
ENV TRANSFORMERS_CACHE=/data/.cache
|
| 7 |
|
| 8 |
WORKDIR /app
|
| 9 |
|
app.py
CHANGED
|
@@ -13,8 +13,6 @@ app.add_middleware(
|
|
| 13 |
allow_headers=["*"],
|
| 14 |
)
|
| 15 |
|
| 16 |
-
os.environ["TRANSFORMERS_CACHE"] = "/data/.cache"
|
| 17 |
-
|
| 18 |
model_name = "pszemraj/long-t5-tglobal-base-16384-book-summary"
|
| 19 |
|
| 20 |
try:
|
|
@@ -22,6 +20,8 @@ try:
|
|
| 22 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
| 23 |
except Exception as e:
|
| 24 |
print("Model loading failed", e)
|
|
|
|
|
|
|
| 25 |
|
| 26 |
class InputText(BaseModel):
|
| 27 |
text: str
|
|
|
|
| 13 |
allow_headers=["*"],
|
| 14 |
)
|
| 15 |
|
|
|
|
|
|
|
| 16 |
model_name = "pszemraj/long-t5-tglobal-base-16384-book-summary"
|
| 17 |
|
| 18 |
try:
|
|
|
|
| 20 |
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
|
| 21 |
except Exception as e:
|
| 22 |
print("Model loading failed", e)
|
| 23 |
+
raise RuntimeError("Failed to load model. Check permissions or cache issues.")
|
| 24 |
+
|
| 25 |
|
| 26 |
class InputText(BaseModel):
|
| 27 |
text: str
|