Update app.py

app.py

@@ -76,7 +76,8 @@ def classify_image(image):
 iface = gr.Interface(
     fn=classify_image,
     inputs=gr.Image(type="pil"),
-    outputs=gr.Textbox(label="Output (Debug Mode)"),  # Use Textbox to display errors if any occur
+    #outputs=gr.Textbox(label="Output (Debug Mode)"),  # Use Textbox to display errors if any occur
+    outputs=gr.Label(num_top_classes=2),
     title="Content Safety Classification",
     description="Upload an image to classify it as 'safe' or 'unsafe' with corresponding probabilities.",
 )