Update app.py

app.py

@@ -61,10 +61,12 @@ def classify_image(image):
         unsafe_prob = probs[0][1].item() * 100  # Unsafe percentage
 
         # Return results
-        return {
+        result = {
             "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}")