Update app.py
Browse files
app.py
CHANGED
@@ -71,7 +71,28 @@ def classify_image(image):
|
|
71 |
image_input = gr.Image(image_mode="RGB") # Removed shape argument
|
72 |
output_text = gr.Textbox()
|
73 |
|
74 |
-
gr.Interface(fn=classify_image, inputs=image_input, outputs=[output_text],
|
75 |
-
|
76 |
-
|
77 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
71 |
image_input = gr.Image(image_mode="RGB") # Removed shape argument
|
72 |
output_text = gr.Textbox()
|
73 |
|
74 |
+
# gr.Interface(fn=classify_image, inputs=image_input, outputs=[output_text],
|
75 |
+
# title="Detect AI-generated Image ",
|
76 |
+
# description="Upload an image to Detected AI-generated Image .",
|
77 |
+
# theme="default").launch()
|
78 |
+
|
79 |
+
gr.Interface(
|
80 |
+
fn=classify_image,
|
81 |
+
inputs=image_input,
|
82 |
+
outputs=[output_text],
|
83 |
+
title="Detect AI-generated Image",
|
84 |
+
description=(
|
85 |
+
"Upload an art image From 6 websites, collecting data from this to detect if it's AI-generated or a real image.\n\n"
|
86 |
+
"### Main Dataset Used:\n"
|
87 |
+
"- [AI-generated Images vs Real Images (Kaggle)](https://www.kaggle.com/datasets/tristanzhang32/ai-generated-images-vs-real-images)\n\n"
|
88 |
+
"**Fake Images Collected From:**\n"
|
89 |
+
"- 10,000 from [Stable Diffusion (OpenArt AI)](https://www.openart.ai)\n"
|
90 |
+
"- 10,000 from [MidJourney (Imagine.Art)](https://www.imagine.art)\n"
|
91 |
+
"- 10,000 from [DALL路E (OpenAI)](https://openai.com/dall-e-2)\n\n"
|
92 |
+
"**Real Images Collected From:**\n"
|
93 |
+
"- 7,500 from [WikiArt](https://www.wikiart.org)\n"
|
94 |
+
"- 22,500 from [Pexels](https://www.pexels.com) and [Unsplash](https://unsplash.com)\n"
|
95 |
+
|
96 |
+
),
|
97 |
+
theme="default"
|
98 |
+
).launch()
|