Update app.py

app.py

@@ -67,6 +67,7 @@ def classify_image(image):
         }
 
     except Exception as e:
+        # Log and return detailed error messages
         print(f"Error during classification: {e}")
         return {"Error": str(e)}
 
@@ -74,13 +75,30 @@ def classify_image(image):
 iface = gr.Interface(
     fn=classify_image,
     inputs=gr.Image(type="pil"),
-    outputs=gr.Label(label="Output"),  # Display probabilities as a percentage scale
+    outputs=gr.Label(label="Output"),  # Display probabilities as progress bars
     title="Content Safety Classification",
     description="Upload an image to classify it as 'safe' or 'unsafe' with corresponding probabilities.",
 )
 
-# Step 4: Launch Gradio Interface
+# Step 4: Add Local Test Before Launching
 if __name__ == "__main__":
+    print("Testing model locally with a sample image...")
+    from PIL import Image
+    import requests
+
+    # Use a sample image
+    url = "https://huggingface.co/datasets/Narsil/image_dummy/raw/main/parrots.png"
+    try:
+        test_image = Image.open(requests.get(url, stream=True).raw)
+
+        # Test the classification function
+        print("Running local test...")
+        test_result = classify_image(test_image)
+        print(f"Local Test Result: {test_result}")
+    except Exception as e:
+        print(f"Error during local test: {e}")
+
+    # Launch Gradio Interface
     print("Launching the Gradio interface...")
     iface.launch()
 
@@ -109,5 +127,6 @@ if __name__ == "__main__":
 
 
 
+