Update Zocket_ImageBind.py

Zocket_ImageBind.py

@@ -49,7 +49,7 @@ if uploaded_image is not None:
 
     softmax_output = torch.softmax(embeddings[ModalityType.VISION] @ embeddings[ModalityType.TEXT].T, dim=-1)
 
-    if softmax_output[0] > softmax_output[1]:
+    if softmax_output[0][0] > softmax_output[0][1]:
         st.write("Advertisement")
     else:
         st.write("Not an Advertisement")