Update app.py

app.py

@@ -2,6 +2,7 @@ import os
 from fastapi import FastAPI, File, UploadFile
 from pydantic import BaseModel
 from transformers import pipeline
+from huggingface_hub import hf_hub_download
 from PIL import Image
 import joblib
 import re
@@ -13,11 +14,9 @@ import uvicorn
 os.environ["TRANSFORMERS_CACHE"] = "/tmp"
 os.environ["HF_HOME"] = "/tmp"
 
-# ✅ Initialize FastAPI
-app = FastAPI()
-
-# ✅ Load NSFW Image Classification Model
+# ✅ Manually Download the NSFW Model to `/tmp`
 try:
+    model_path = hf_hub_download(repo_id="LukeJacob2023/nsfw-image-detector", filename="pytorch_model.bin", cache_dir="/tmp")
     pipe = pipeline("image-classification", model="LukeJacob2023/nsfw-image-detector", cache_dir="/tmp")
     print("✅ NSFW Model Loaded Successfully!")
 except Exception as e:
@@ -33,6 +32,9 @@ except Exception as e:
     print(f"❌ Error Loading Toxic Text Model: {e}")
     exit(1)
 
+# ✅ Initialize FastAPI
+app = FastAPI()
+
 # 📌 Text Input Model
 class TextInput(BaseModel):
     text: str