Emilyl613 commited on
Commit
d1dc6dc
·
verified ·
1 Parent(s): 143097c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -142,8 +142,10 @@ with gr.Blocks(title="AI Image Detector") as demo:
142
  with gr.Row():
143
  upload_input = gr.Image(label="Upload Image")
144
  with gr.Column():
145
- upload_our = gr.Markdown(label="🔵 Our CNN (trained on CIFAKE 32×32)")
146
- upload_pretrained = gr.Markdown(label="🟡 Pretrained Detector (Organika/sdxl-detector)")
 
 
147
  upload_gradcam = gr.Image(label="GradCAM (Our CNN)")
148
  upload_btn = gr.Button("Detect", variant="primary")
149
  upload_btn.click(detect_image, inputs=upload_input,
@@ -156,8 +158,10 @@ with gr.Blocks(title="AI Image Detector") as demo:
156
  with gr.Row():
157
  generated_img = gr.Image(label="Generated Image")
158
  with gr.Column():
159
- generate_our = gr.Markdown(label="🔵 Our CNN (trained on CIFAKE 32×32)")
160
- generate_pretrained = gr.Markdown(label="🟡 Pretrained Detector (Organika/sdxl-detector)")
 
 
161
  generate_gradcam = gr.Image(label="GradCAM (Our CNN)")
162
  generate_btn.click(generate_and_detect, inputs=prompt_input,
163
  outputs=[generated_img, generate_our, generate_pretrained, generate_gradcam])
 
142
  with gr.Row():
143
  upload_input = gr.Image(label="Upload Image")
144
  with gr.Column():
145
+ gr.Markdown("**🔵 Our CNN (trained on CIFAKE 32×32)**")
146
+ upload_our = gr.Markdown()
147
+ gr.Markdown("**🟡 Pretrained Detector (Organika/sdxl-detector)**")
148
+ upload_pretrained = gr.Markdown()
149
  upload_gradcam = gr.Image(label="GradCAM (Our CNN)")
150
  upload_btn = gr.Button("Detect", variant="primary")
151
  upload_btn.click(detect_image, inputs=upload_input,
 
158
  with gr.Row():
159
  generated_img = gr.Image(label="Generated Image")
160
  with gr.Column():
161
+ gr.Markdown("**🔵 Our CNN (trained on CIFAKE 32×32)**")
162
+ generate_our = gr.Markdown()
163
+ gr.Markdown("**🟡 Pretrained Detector (Organika/sdxl-detector)**")
164
+ generate_pretrained = gr.Markdown()
165
  generate_gradcam = gr.Image(label="GradCAM (Our CNN)")
166
  generate_btn.click(generate_and_detect, inputs=prompt_input,
167
  outputs=[generated_img, generate_our, generate_pretrained, generate_gradcam])