Update app.py

app.py

@@ -142,8 +142,8 @@ with gr.Blocks(title="AI Image Detector") as demo:
             with gr.Row():
                 upload_input = gr.Image(label="Upload Image")
                 with gr.Column():
-                    upload_our = gr.Markdown(label="Our CNN (CIFAKE)")
-                    upload_pretrained = gr.Markdown(label="Pretrained Detector")
+                    upload_our = gr.Markdown(label="🔵 Our CNN (trained on CIFAKE 32×32)")
+                    upload_pretrained = gr.Markdown(label="🟡 Pretrained Detector (Organika/sdxl-detector)")
                     upload_gradcam = gr.Image(label="GradCAM (Our CNN)")
             upload_btn = gr.Button("Detect", variant="primary")
             upload_btn.click(detect_image, inputs=upload_input,
@@ -156,8 +156,8 @@ with gr.Blocks(title="AI Image Detector") as demo:
             with gr.Row():
                 generated_img = gr.Image(label="Generated Image")
                 with gr.Column():
-                    generate_our = gr.Markdown(label="Our CNN (CIFAKE)")
-                    generate_pretrained = gr.Markdown(label="Pretrained Detector")
+                    generate_our = gr.Markdown(label="🔵 Our CNN (trained on CIFAKE 32×32)")
+                    generate_pretrained = gr.Markdown(label="🟡 Pretrained Detector (Organika/sdxl-detector)")
                     generate_gradcam = gr.Image(label="GradCAM (Our CNN)")
             generate_btn.click(generate_and_detect, inputs=prompt_input,
                              outputs=[generated_img, generate_our, generate_pretrained, generate_gradcam])