Update main.py

main.py

@@ -1,15 +1,20 @@
+import os
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf-cache"
+os.environ["HF_HOME"] = "/tmp/hf-home"
+
 from fastapi import FastAPI, Request
 from pydantic import BaseModel
 from transformers import pipeline
 
+# Define FastAPI app
+app = FastAPI()
+
 # Load zero-shot classification pipeline
 classifier = pipeline(
     "zero-shot-classification",
     model="MoritzLaurer/deberta-v3-large-zeroshot-v2.0"
 )
 
-# Define FastAPI app
-app = FastAPI()
 
 # Define input schema
 class InputText(BaseModel):