From 59e49a36ce8a64b4591cce32a62a649e44d19af1 Mon Sep 17 00:00:00 2001 From: wxywb Date: Tue, 11 Oct 2022 10:20:00 +0800 Subject: [PATCH] update the taiyi requirement. Signed-off-by: wxywb --- requirements.txt | 3 +++ taiyi.py | 5 ----- 2 files changed, 3 insertions(+), 5 deletions(-) create mode 100644 requirements.txt diff --git a/requirements.txt b/requirements.txt new file mode 100644 index 0000000..1d76662 --- /dev/null +++ b/requirements.txt @@ -0,0 +1,3 @@ +torch +transformers +towhee diff --git a/taiyi.py b/taiyi.py index 27123a5..2f901ab 100644 --- a/taiyi.py +++ b/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