Spaces:
Sleeping
Sleeping
Commit
·
437ec30
1
Parent(s):
37d1dae
Fix: argument mismatch
Browse files
app.py
CHANGED
@@ -30,11 +30,11 @@ interface = gr.Interface(
|
|
30 |
type="pil",
|
31 |
label="Upload an Image",
|
32 |
interactive=True,
|
33 |
-
value=None
|
34 |
),
|
35 |
gr.CheckboxGroup(
|
36 |
choices=["yolov5", "yolov8s"],
|
37 |
-
value=["yolov5"], #
|
38 |
label="Select Model(s)",
|
39 |
type="value", # Ensure the value is passed as a list of selected models
|
40 |
),
|
@@ -43,12 +43,15 @@ interface = gr.Interface(
|
|
43 |
title="Explainable AI for YOLO Models",
|
44 |
description="Upload an image or select a sample to visualize YOLO object detection with Grad-CAM.",
|
45 |
examples=[
|
46 |
-
[os.path.join(os.getcwd(), "data/xai/sample1.jpeg")
|
47 |
-
[os.path.join(os.getcwd(), "data/xai/sample2.jpg")
|
48 |
],
|
49 |
-
live=True
|
|
|
50 |
)
|
51 |
|
52 |
|
|
|
|
|
53 |
if __name__ == "__main__":
|
54 |
interface.launch()
|
|
|
30 |
type="pil",
|
31 |
label="Upload an Image",
|
32 |
interactive=True,
|
33 |
+
value=None,
|
34 |
),
|
35 |
gr.CheckboxGroup(
|
36 |
choices=["yolov5", "yolov8s"],
|
37 |
+
value=["yolov5"], # Default value set
|
38 |
label="Select Model(s)",
|
39 |
type="value", # Ensure the value is passed as a list of selected models
|
40 |
),
|
|
|
43 |
title="Explainable AI for YOLO Models",
|
44 |
description="Upload an image or select a sample to visualize YOLO object detection with Grad-CAM.",
|
45 |
examples=[
|
46 |
+
[os.path.join(os.getcwd(), "data/xai/sample1.jpeg")],
|
47 |
+
[os.path.join(os.getcwd(), "data/xai/sample2.jpg")],
|
48 |
],
|
49 |
+
live=True,
|
50 |
+
allow_flagging="never", # Optional: Disable flagging if not required
|
51 |
)
|
52 |
|
53 |
|
54 |
+
|
55 |
+
|
56 |
if __name__ == "__main__":
|
57 |
interface.launch()
|