logo
Browse Source

amend the requirement.

Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb 2 years ago
parent
commit
dd8e14ef07
  1. 4
      README.md
  2. 3
      requirements.txt

4
README.md

@ -64,14 +64,14 @@ Create the operator via the following factory method
## Interface
An image-text embedding operator takes a [towhee image](link/to/towhee/image/api/doc) as input and generate the correspoing caption.
An image captioning operator takes a [towhee image](link/to/towhee/image/api/doc) as input and generate the correspoing caption.
**Parameters:**
***img:*** *towhee.types.Image (a sub-class of numpy.ndarray)*
​ The image to generate embedding.
​ The image to generate caption.

3
requirements.txt

@ -3,3 +3,6 @@ torch
torchvision
timm
towhee
ftfy
yacs
numpy

Loading…
Cancel
Save