Update app.py

app.py

@@ -61,12 +61,10 @@ def classify_image(image):
         unsafe_prob = probs[0][1].item() * 100  # Unsafe percentage
 
         # Return results
-        result = {
+        return {
             "safe": f"{safe_prob:.2f}%",
             "unsafe": f"{unsafe_prob:.2f}%"
         }
-        return dict(sorted(result.items(), key=lambda item: item[1], reverse=True))
-
 
     except Exception as e:
         print(f"Error during classification: {e}")
@@ -76,8 +74,7 @@ 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.Label(num_top_classes=2),
+    outputs=gr.Textbox(label="Output (Debug Mode)"),  # Use Textbox to display errors if any occur
     title="Content Safety Classification",
     description="Upload an image to classify it as 'safe' or 'unsafe' with corresponding probabilities.",
 )
@@ -119,5 +116,6 @@ if __name__ == "__main__":
 
 
 
+