Update app.py

app.py

@@ -1,3 +1,4 @@
+import os
 from fastapi import FastAPI, File, UploadFile
 from pydantic import BaseModel
 from transformers import pipeline
@@ -6,36 +7,33 @@ import joblib
 import re
 import string
 import io
-import os
 import uvicorn
-import os
-# ✅ Set Hugging Face Cache Directory (Fixes Permission Error)
-import os
-
-# ✅ Set Hugging Face Cache to a writable directory
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
-os.environ["HF_HOME"] = "/tmp/hf_cache"
-
-
-
 
+# ✅ Set Hugging Face Cache to a writable directory (Fixes Permission Error)
+os.environ["TRANSFORMERS_CACHE"] = "/tmp"
+os.environ["HF_HOME"] = "/tmp"
 
 # ✅ Initialize FastAPI
 app = FastAPI()
-pipe = pipeline("image-classification", model="LukeJacob2023/nsfw-image-detector",cache_dir="/tmp")
-# ✅ Load NSFW Image Classification Model (with custom cache directory)
 
+# ✅ Load NSFW Image Classification Model
+try:
+    pipe = pipeline("image-classification", model="LukeJacob2023/nsfw-image-detector", cache_dir="/tmp")
+    print("✅ NSFW Model Loaded Successfully!")
+except Exception as e:
+    print(f"❌ Error Loading NSFW Model: {e}")
+    exit(1)
 
 # ✅ Load Toxic Text Classification Model
 try:
     model = joblib.load("toxic_classifier.pkl")
     vectorizer = joblib.load("vectorizer.pkl")
-    print("✅ Model & Vectorizer Loaded Successfully!")
+    print("✅ Toxic Text Model & Vectorizer Loaded Successfully!")
 except Exception as e:
-    print(f"❌ Error: {e}")
+    print(f"❌ Error Loading Toxic Text Model: {e}")
     exit(1)
 
-# 📌 Text Input Data Model
+# 📌 Text Input Model
 class TextInput(BaseModel):
     text: str