Update app.py

app.py

@@ -90,10 +90,9 @@ model.eval()
 
 detoxify_model = Detoxify('multilingual')
 
-# Use a more accurate NSFW image detection model
+# Use the Falconsai NSFW image detection model
 print("Loading NSFW image classification model...")
-# Use a smaller, faster model for NSFW detection
-nsfw_classifier = pipeline("image-classification", model="cafeai/nsfw-detector")
+nsfw_classifier = pipeline("image-classification", model="Falconsai/nsfw_image_detection")
 print("NSFW image classification model loaded.")
 
 MODERATION_SYSTEM_PROMPT = (
@@ -306,10 +305,7 @@ def classify_text_with_detoxify(text):
 def classify_image(image_data):
     try:
         img = Image.open(io.BytesIO(image_data)).convert("RGB")
-        # Resize for faster processing
-        img = img.resize((224, 224))
-        
-        # Use the NSFW detector
+        # Use the Falconsai NSFW detector
         results = nsfw_classifier(img)
         
         # Extract the NSFW score
@@ -319,8 +315,8 @@ def classify_image(image_data):
                 nsfw_score = result['score']
                 break
         
-        # Use a threshold of 0.7 for stricter detection
-        classification = 'u' if nsfw_score > 0.7 else 's'
+        # Use a threshold of 0.5 for NSFW detection
+        classification = 'u' if nsfw_score > 0.5 else 's'
         
         return {
             "classification": classification,