ImageBind_zeroshot_demo

Runtime error

JustinLin610 commited on May 11, 2023

Commit

fa7d0e2

•

1 Parent(s): 777a816

debug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,7 @@ def image_text_zeroshot(image, text_list):
         embeddings = model(inputs)
     scores = torch.softmax(
-        embeddings[ModalityType.VISION] @ embeddings[ModalityType.AUDIO].T,
         dim=-1
     ).squeeze(0).tolist()

         embeddings = model(inputs)
     scores = torch.softmax(
+        embeddings[ModalityType.VISION] @ embeddings[ModalityType.TEXT].T,
         dim=-1
     ).squeeze(0).tolist()