Spaces:
Sleeping
Sleeping
ai >>> AI Generated
#1
by
lunarflu
HF staff
- opened
app.py
CHANGED
|
@@ -27,7 +27,7 @@ def main(img):
|
|
| 27 |
if predictions[0][0] > confidence:
|
| 28 |
class_label = 'real'
|
| 29 |
else:
|
| 30 |
-
class_label = '
|
| 31 |
|
| 32 |
# Clean up the temporary image file
|
| 33 |
os.remove(img_path)
|
|
|
|
| 27 |
if predictions[0][0] > confidence:
|
| 28 |
class_label = 'real'
|
| 29 |
else:
|
| 30 |
+
class_label = 'AI Generated'
|
| 31 |
|
| 32 |
# Clean up the temporary image file
|
| 33 |
os.remove(img_path)
|