Update app.py

app.py

@@ -60,10 +60,10 @@ def classify_image(image):
         safe_prob = probs[0][0].item() * 100  # Safe percentage
         unsafe_prob = probs[0][1].item() * 100  # Unsafe percentage
 
-        # Return results
+        # Format data for visualization
         return {
-            "safe": f"{safe_prob:.2f}%",
-            "unsafe": f"{unsafe_prob:.2f}%"
+            "safe": safe_prob,
+            "unsafe": unsafe_prob
         }
 
     except Exception as e:
@@ -74,7 +74,7 @@ def classify_image(image):
 iface = gr.Interface(
     fn=classify_image,
     inputs=gr.Image(type="pil"),
-    outputs=gr.Text(label="Output"),  
+    outputs=gr.Label(num_top_classes=2),  # Use gr.Label to display probabilities with a bar-style visualization
     title="Content Safety Classification",
     description="Upload an image to classify it as 'safe' or 'unsafe' with corresponding probabilities.",
 )
@@ -117,5 +117,6 @@ if __name__ == "__main__":
 
 
 
+