logo
Browse Source

update the taiyi requirement.

Signed-off-by: wxywb <xy.wang@zilliz.com>
main
wxywb 2 years ago
parent
commit
59e49a36ce
  1. 3
      requirements.txt
  2. 5
      taiyi.py

3
requirements.txt

@ -0,0 +1,3 @@
torch
transformers
towhee

5
taiyi.py

@ -15,11 +15,6 @@ import torch
from transformers import BertForSequenceClassification, BertConfig, BertTokenizer
from transformers import CLIPProcessor, CLIPModel
import sys
from pathlib import Path
import torch
from torchvision import transforms
from towhee.types.image_utils import to_pil
from towhee.operator.base import NNOperator, OperatorFlag
from towhee.types.arg import arg, to_image_color

Loading…
Cancel
Save