noumanjavaid commited on
Commit
36a2325
·
verified ·
1 Parent(s): 6188830

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -66,6 +66,7 @@
66
 
67
  # # Process the image
68
  # with gr.spinner("Analyzing image..."):
 
69
  # result = process_image(img_byte_arr, api_key)
70
 
71
  # if result and result['status'] == 'SUCCESS':
@@ -151,7 +152,7 @@ def process_image(image):
151
 
152
  if 'data' in result and len(result['data']) > 0:
153
  first_result = result['data'][0]
154
- if 'is_ai_generated' in first_result:
155
  confidence = first_result['is_ai_generated']
156
  sources = first_result.get('possible_sources', {})
157
 
@@ -161,7 +162,7 @@ def process_image(image):
161
  result_message += f"\n Confidence score: {confidence:.2%}\n\n"
162
 
163
  if is_ai_generated == "Yes" and sources:
164
- result_message += "\n Top possible AI models used:\n"
165
  sorted_sources = sorted(sources.items(), key=lambda x: x[1], reverse=True)[:3]
166
  for source, prob in sorted_sources:
167
  if prob > 0.01:
 
66
 
67
  # # Process the image
68
  # with gr.spinner("Analyzing image..."):
69
+
70
  # result = process_image(img_byte_arr, api_key)
71
 
72
  # if result and result['status'] == 'SUCCESS':
 
152
 
153
  if 'data' in result and len(result['data']) > 0:
154
  first_result = result['data'][0]
155
+ if '**is_ai_generated**' in first_result:
156
  confidence = first_result['is_ai_generated']
157
  sources = first_result.get('possible_sources', {})
158
 
 
162
  result_message += f"\n Confidence score: {confidence:.2%}\n\n"
163
 
164
  if is_ai_generated == "Yes" and sources:
165
+ result_message += "\n **Top possible AI models used** :\n"
166
  sorted_sources = sorted(sources.items(), key=lambda x: x[1], reverse=True)[:3]
167
  for source, prob in sorted_sources:
168
  if prob > 0.01: