Spaces:

vinid
/

webplip

Running

webplip / plip_support.py

Upload plip_support.py

21847fd over 1 year ago

No virus

187 Bytes

	import clip
	import torch




	def embed_text(plip, text, device="cpu"):
	idx = clip.tokenize([text], truncate=True).to(device)
	return plip.encode_text(idx).detach().cpu().numpy()[0]