Spaces:
Runtime error
Runtime error
JustinLin610
commited on
Commit
•
777a816
1
Parent(s):
1bb90fa
debug
Browse files
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),
|