Thanaphit commited on
Commit
59632e2
1 Parent(s): a6e2c31

Add examples for damage severity

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. 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
- # examples=SAMPLE["car-dam-det"],
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):