Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 from fastapi import FastAPI, UploadFile, File
 from transformers import AutoModelForImageClassification, AutoImageProcessor
 from PIL import Image
@@ -5,6 +6,9 @@ import torch.nn.functional as F
 import torch
 import io
 
+# ✅ Fix: set writable cache directory
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface"
+
 app = FastAPI()
 
 model = AutoModelForImageClassification.from_pretrained("prithivMLmods/Gender-Classifier-Mini")