Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -158,14 +158,14 @@ def process_image(image):
|
|
158 |
sources = first_result.get('possible_sources', {})
|
159 |
|
160 |
# Format the result message
|
161 |
-
is_ai_generated = "Yes" if confidence >= 0.5 else "No"
|
162 |
-
result_message = f"\n Is this image AI-generated? {is_ai_generated}\n"
|
163 |
result_message += f"\n Confidence score: {confidence:.2%}\n\n"
|
164 |
-
|
165 |
if is_ai_generated == "Yes" and sources:
|
166 |
result_message += "\n **Top possible AI models used** :\n"
|
167 |
sorted_sources = sorted(sources
|
168 |
-
.items(), key=lambda x: x[1], reverse=True)[:3]
|
169 |
for source, prob in sorted_sources:
|
170 |
if prob > 0.01:
|
171 |
result_message += f"- {source}: {prob:.1%}\n"
|
|
|
158 |
sources = first_result.get('possible_sources', {})
|
159 |
|
160 |
# Format the result message
|
161 |
+
is_ai_generated = "### <span style='color:red'>Yes</span>" if confidence >= 0.5 else "No"
|
162 |
+
result_message = f"\n ### Is this image AI-generated? {is_ai_generated}\n"
|
163 |
result_message += f"\n Confidence score: {confidence:.2%}\n\n"
|
164 |
+
|
165 |
if is_ai_generated == "Yes" and sources:
|
166 |
result_message += "\n **Top possible AI models used** :\n"
|
167 |
sorted_sources = sorted(sources
|
168 |
+
.items(), key=lambda x: x[1], reverse=True)[:3]
|
169 |
for source, prob in sorted_sources:
|
170 |
if prob > 0.01:
|
171 |
result_message += f"- {source}: {prob:.1%}\n"
|