소스 검색

add img embedding.py

luojunhui 3 주 전
부모
커밋
d75687a4fb
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      applications/clip_embedding/embedding.py

+ 1 - 1
applications/clip_embedding/embedding.py

@@ -37,6 +37,6 @@ async def embed_image_url(img_url_list: List[str]):
             image_features = model.encode_image(pixel_values)
 
         feats = _normalize(image_features)
-        outputs.extend(_to_list(image_features))
+        outputs.extend(_to_list(feats))
 
     return outputs