Upload folder using huggingface_hub

README.md

@@ -72,7 +72,7 @@ image = Image.open(path)
 # Extract UNI embedding from the image
 uni_inp = transform(image).unsqueeze(dim=0)
 with torch.inference_mode():
-    uni_emb = uni_model(uni_inp.to(device)).unsqueeze(dim=0)
+    uni_emb = uni_model(uni_inp.to(device))
 
 print("Extracted UNI:", uni_emb.shape)