Spaces:
Runtime error
Runtime error
Add examples for damage severity
Browse files- app.py +1 -1
- utils/__init__.py +3 -0
app.py
CHANGED
@@ -42,7 +42,7 @@ car_sev_ui = gr.Interface(
|
|
42 |
],
|
43 |
outputs=gr.components.Image(type="numpy", label="Output Image"),
|
44 |
title="Car Damage Severity Detector",
|
45 |
-
|
46 |
cache_examples=False,
|
47 |
)
|
48 |
|
|
|
42 |
],
|
43 |
outputs=gr.components.Image(type="numpy", label="Output Image"),
|
44 |
title="Car Damage Severity Detector",
|
45 |
+
examples=SAMPLE["car-dam-sev-det"],
|
46 |
cache_examples=False,
|
47 |
)
|
48 |
|
utils/__init__.py
CHANGED
@@ -23,6 +23,9 @@ def setup():
|
|
23 |
"car-dam-det" : [ [f"{root}/{file}"] \
|
24 |
for root, _, files in os.walk("sample/car-damage-det", topdown=False) \
|
25 |
for file in files ],
|
|
|
|
|
|
|
26 |
}
|
27 |
|
28 |
if not os.path.exists(CAR_PART_SEG_OUT):
|
|
|
23 |
"car-dam-det" : [ [f"{root}/{file}"] \
|
24 |
for root, _, files in os.walk("sample/car-damage-det", topdown=False) \
|
25 |
for file in files ],
|
26 |
+
"car-dam-sev-det" : [ [f"{root}/{file}"] \
|
27 |
+
for root, _, files in os.walk("sample/car-damage-sev-det", topdown=False) \
|
28 |
+
for file in files ],
|
29 |
}
|
30 |
|
31 |
if not os.path.exists(CAR_PART_SEG_OUT):
|