Update app.py

app.py

@@ -1,3 +1,12 @@
+import os
+
+# =============================
+# Fix cache permissions pour HF
+# =============================
+os.environ["HF_HOME"] = "/code/.cache"
+os.environ["TRANSFORMERS_CACHE"] = "/code/.cache"
+os.environ["HF_HUB_CACHE"] = "/code/.cache"
+
 from fastapi import FastAPI
 from pydantic import BaseModel
 from transformers import AutoTokenizer, AutoModelForCausalLM, AutoModelForSequenceClassification
@@ -156,3 +165,10 @@ class RequestBody(BaseModel):
 @app.post("/predict")
 async def predict_api(body: RequestBody):
     return classify_and_respond(body.text)
+
+# =============================
+# Lancement local (uvicorn)
+# =============================
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=7860)