Spaces:
Running
Running
Update app.py
Browse files
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:
|