noumanjavaid commited on
Commit
81244e1
·
verified ·
1 Parent(s): 2c56a67

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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"