showing labels

Zocket_ImageBind.py

@@ -47,9 +47,9 @@ if uploaded_image is not None:
     with torch.no_grad():
         embeddings = model(inputs)
 
-    print(
-        "Vision x Text: ",
-        torch.softmax(embeddings[ModalityType.VISION] @ embeddings[ModalityType.TEXT].T, dim=-1),
-    )
+    softmax_output = torch.softmax(embeddings[ModalityType.VISION] @ embeddings[ModalityType.TEXT].T, dim=-1)
 
-    st.write(torch.softmax(embeddings[ModalityType.VISION] @ embeddings[ModalityType.TEXT].T, dim=-1))
+    if softmax_output[0] > softmax_output[1]:
+        st.write("Advertisement")
+    else:
+        st.write("Not an Advertisement")