JustinLin610 commited on
Commit
3388bb6
1 Parent(s): 7b9d1bc
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ def image_text_zeroshot(image, text_list):
15
  image_paths = [image]
16
  labels = [label.strip(" ") for label in text_list.strip(" ").split("|")]
17
  inputs = {
18
- ModalityType.TEXT: data.load_and_transform_text(text_list, device),
19
  ModalityType.VISION: data.load_and_transform_vision_data(image_paths, device),
20
  }
21
 
 
15
  image_paths = [image]
16
  labels = [label.strip(" ") for label in text_list.strip(" ").split("|")]
17
  inputs = {
18
+ ModalityType.TEXT: data.load_and_transform_text(labels, device),
19
  ModalityType.VISION: data.load_and_transform_vision_data(image_paths, device),
20
  }
21