JustinLin610 commited on
Commit
777a816
1 Parent(s): 1bb90fa
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ model.to(device)
13
 
14
  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),
 
13
 
14
  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),