ImageBind_zeroshot_demo2

Runtime error

Rajagopal commited on Aug 4, 2023

Commit

ecf0e06

1 Parent(s): b68d4a2

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -93,7 +93,7 @@ def doubleimage_text_zeroshot(image, image2,  text_list):
     with torch.no_grad():
         embeddings = model(inputs)
-    )
     return        str(torch.softmax(embeddings[ModalityType.VISION] @ embeddings[ModalityType.TEXT].T, dim=-1)   )

     with torch.no_grad():
         embeddings = model(inputs)
     return        str(torch.softmax(embeddings[ModalityType.VISION] @ embeddings[ModalityType.TEXT].T, dim=-1)   )