cfoli commited on
Commit
081aab9
·
verified ·
1 Parent(s): 9108ad3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -42,6 +42,7 @@ configs = {
42
  "STD": (0.229, 0.224, 0.225),
43
 
44
  "DEFAULT_BACKBONE": "CheXFormer-small",
 
45
 
46
  "THRESHOLD": 0.2
47
  }
@@ -363,7 +364,7 @@ example_list_dir = os.path.join(os.getcwd(), "Curated test samples")
363
  example_list_img_names = os.listdir(example_list_dir)
364
 
365
  example_list = [
366
- [configs["DEFAULT_BACKBONE"], os.path.join(example_list_dir, example_img)]
367
  for example_img in example_list_img_names[:8]
368
  if example_img.lower().endswith(".png")]
369
 
 
42
  "STD": (0.229, 0.224, 0.225),
43
 
44
  "DEFAULT_BACKBONE": "CheXFormer-small",
45
+ "DEFAULT_VLM": "Lingshu-7B",
46
 
47
  "THRESHOLD": 0.2
48
  }
 
364
  example_list_img_names = os.listdir(example_list_dir)
365
 
366
  example_list = [
367
+ [configs["DEFAULT_BACKBONE"], configs["DEFAULT_VLM"], os.path.join(example_list_dir, example_img)]
368
  for example_img in example_list_img_names[:8]
369
  if example_img.lower().endswith(".png")]
370