Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import CLIPModel, CLIPProcessor
+from PIL import Image
 
 # Step 1: Load Fine-Tuned Model from Hugging Face Model Hub
 model_name = "quadranttechnologies/retail-content-safety-clip-finetuned"
@@ -21,7 +22,7 @@ def classify_image(image):
     Classify an image as 'safe' or 'unsafe' and return probabilities.
 
     Args:
-        image (PIL.Image.Image): The uploaded image.
+        image (PIL.Image.Image): Uploaded image.
     
     Returns:
         str: Predicted category ("safe" or "unsafe").
@@ -105,5 +106,6 @@ if __name__ == "__main__":
 
 
 
+