Spaces:
Sleeping
Sleeping
Commit
·
408a665
1
Parent(s):
084f1b5
Update: background template
Browse files
app.py
CHANGED
@@ -46,49 +46,48 @@ def process_image(sample_choice, uploaded_image, yolo_versions=["yolov5"]):
|
|
46 |
result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
|
47 |
return result_images
|
48 |
|
49 |
-
with gr.Blocks(css=".
|
50 |
-
|
51 |
-
|
52 |
-
|
53 |
-
|
54 |
-
|
55 |
-
|
56 |
-
|
57 |
-
|
58 |
-
|
59 |
-
|
60 |
-
|
61 |
-
value=default_sample, # Set default selection
|
62 |
-
)
|
63 |
-
# Upload image below sample selection
|
64 |
-
gr.Markdown("**Or upload your own image:**")
|
65 |
-
upload_image = gr.Image(
|
66 |
-
label="Upload an Image",
|
67 |
-
type="pil", # Correct type for file path compatibility
|
68 |
-
)
|
69 |
-
# Right side: Selected sample image display
|
70 |
-
sample_display = gr.Image(
|
71 |
-
value=load_sample_image(default_sample),
|
72 |
-
label="Selected Sample Image",
|
73 |
)
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
|
|
|
|
|
|
|
|
|
|
79 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
80 |
|
81 |
-
|
82 |
-
|
83 |
-
|
84 |
-
|
85 |
-
|
86 |
-
|
87 |
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
|
94 |
interface.launch(share=True)
|
|
|
46 |
result_images.append((Image.fromarray(image), f"{yolo_version} not yet implemented."))
|
47 |
return result_images
|
48 |
|
49 |
+
with gr.Blocks(css=".orchid-green-bg { background-color: #32CD32 !important; padding: 20px; }") as interface:
|
50 |
+
gr.Markdown("# XAI: Visualize Object Detection of Your Models")
|
51 |
+
gr.Markdown("Select a sample image to visualize object detection.")
|
52 |
+
default_sample = "Sample 1"
|
53 |
+
with gr.Row(elem_classes="orchid-green-bg"):
|
54 |
+
# Left side: Sample selection and upload image
|
55 |
+
with gr.Column():
|
56 |
+
sample_selection = gr.Radio(
|
57 |
+
choices=list(sample_images.keys()),
|
58 |
+
label="Select a Sample Image",
|
59 |
+
type="value",
|
60 |
+
value=default_sample, # Set default selection
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
61 |
)
|
62 |
+
# Upload image below sample selection
|
63 |
+
gr.Markdown("**Or upload your own image:**")
|
64 |
+
upload_image = gr.Image(
|
65 |
+
label="Upload an Image",
|
66 |
+
type="pil", # Correct type for file path compatibility
|
67 |
+
)
|
68 |
+
# Right side: Selected sample image display
|
69 |
+
sample_display = gr.Image(
|
70 |
+
value=load_sample_image(default_sample),
|
71 |
+
label="Selected Sample Image",
|
72 |
)
|
73 |
+
|
74 |
+
sample_selection.change(
|
75 |
+
fn=load_sample_image,
|
76 |
+
inputs=sample_selection,
|
77 |
+
outputs=sample_display,
|
78 |
+
)
|
79 |
|
80 |
+
selected_models = gr.CheckboxGroup(
|
81 |
+
choices=["yolov5", "yolov8s"],
|
82 |
+
value=["yolov5"],
|
83 |
+
label="Select Model(s)",
|
84 |
+
)
|
85 |
+
result_gallery = gr.Gallery(label="Results", elem_id="gallery", rows=2, height=500)
|
86 |
|
87 |
+
gr.Button("Run").click(
|
88 |
+
fn=process_image,
|
89 |
+
inputs=[sample_selection, upload_image, selected_models], # Include both options
|
90 |
+
outputs=result_gallery,
|
91 |
+
)
|
92 |
|
93 |
interface.launch(share=True)
|