MichalMlodawski
commited on
Commit
•
89b1df6
1
Parent(s):
c987077
Update app.py
Browse files
app.py
CHANGED
@@ -52,8 +52,8 @@ def classify_image(image):
|
|
52 |
emoji = {"Safe": "✅", "Questionable": "⚠️", "Unsafe": "🔞"}.get(category, "❓")
|
53 |
confidence_bar = "🟩" * int(confidence_value // 10) + "⬜" * (10 - int(confidence_value // 10))
|
54 |
|
55 |
-
result = f"{emoji} NSFW Category: {category}\n"
|
56 |
-
result
|
57 |
result += f"🎯 Confidence: {confidence_value:.2f}% {confidence_bar}"
|
58 |
|
59 |
return result
|
|
|
52 |
emoji = {"Safe": "✅", "Questionable": "⚠️", "Unsafe": "🔞"}.get(category, "❓")
|
53 |
confidence_bar = "🟩" * int(confidence_value // 10) + "⬜" * (10 - int(confidence_value // 10))
|
54 |
|
55 |
+
#result = f"{emoji} NSFW Category: {category}\n"
|
56 |
+
result = f"🏷️ Model Label: {label}\n"
|
57 |
result += f"🎯 Confidence: {confidence_value:.2f}% {confidence_bar}"
|
58 |
|
59 |
return result
|