Spaces:
Sleeping
Sleeping
Commit
·
66370da
1
Parent(s):
f7b8e0e
Fix: model dependency + default sample
Browse files
app.py
CHANGED
@@ -26,6 +26,8 @@ sample_images = {
|
|
26 |
"Sample 1": os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
|
27 |
"Sample 2": os.path.join(os.getcwd(), "data/xai/sample2.jpg"),
|
28 |
}
|
|
|
|
|
29 |
def load_sample_image(sample_name):
|
30 |
if sample_name in sample_images:
|
31 |
try:
|
@@ -46,11 +48,11 @@ with gr.Blocks() as interface:
|
|
46 |
label="Select a Sample Image",
|
47 |
type="value",
|
48 |
)
|
49 |
-
sample_display = gr.Image(label="Sample Image Preview",
|
50 |
sample_selection.change(fn=load_sample_image, inputs=sample_selection, outputs=sample_display)
|
51 |
|
52 |
selected_models = gr.CheckboxGroup(
|
53 |
-
choices=["
|
54 |
value=["yolov5"], # Default model
|
55 |
label="Select Model(s)",
|
56 |
)
|
|
|
26 |
"Sample 1": os.path.join(os.getcwd(), "data/xai/sample1.jpeg"),
|
27 |
"Sample 2": os.path.join(os.getcwd(), "data/xai/sample2.jpg"),
|
28 |
}
|
29 |
+
default_sample_image = load_sample_image("Sample 1")
|
30 |
+
|
31 |
def load_sample_image(sample_name):
|
32 |
if sample_name in sample_images:
|
33 |
try:
|
|
|
48 |
label="Select a Sample Image",
|
49 |
type="value",
|
50 |
)
|
51 |
+
sample_display = gr.Image(label="Sample Image Preview", value=default_sample_image)
|
52 |
sample_selection.change(fn=load_sample_image, inputs=sample_selection, outputs=sample_display)
|
53 |
|
54 |
selected_models = gr.CheckboxGroup(
|
55 |
+
choices=["yolov5", "yolov8s"],
|
56 |
value=["yolov5"], # Default model
|
57 |
label="Select Model(s)",
|
58 |
)
|