Update app.py

app.py

@@ -78,6 +78,7 @@ moderation_model = pipeline(
     model="Vrandan/Comment-Moderation",
     return_all_scores=True
 )
+
 from PIL import Image
 import torch
 from gtts import gTTS
@@ -117,20 +118,17 @@ print("✅ All models loaded!")
 # Safety Filter Function
 # ----------------------
 def is_caption_safe(caption):
-    # Run the moderation model to get scores
     votes = moderation_model(caption)[0]
-    # Check for violence/graphic violence labels
     for item in votes:
-        if item["label"] in ["V", "V2"] and item["score"] > 0.5:
+        if item["label"] in ["V", "V2"] and item["score"] > 0.5:  # Violence / Graphic Violence
             return False
-    # Additionally flag via keywords
-    unsafe_keywords = [
-        "gun", "blood", "skull", "kill", "corpse", "gore", "knife", "weapon"
-    ]
+    unsafe_keywords = ["gun", "blood", "skull", "kill", "corpse", "gore", "knife", "weapon"]
     if any(word in caption.lower() for word in unsafe_keywords):
         return False
     return True
 
+
+
 # ----------------------
 # Caption + Translate + Speak
 # ----------------------