From 4da37ac8c624460c57727149899f67f7c8c97dcc Mon Sep 17 00:00:00 2001
From: wxywb <xy.wang@zilliz.com>
Date: Thu, 13 Oct 2022 14:26:25 +0800
Subject: [PATCH] update the requirement.

Signed-off-by: wxywb <xy.wang@zilliz.com>
---
 clip_caption_reward.py | 1 -
 requirements.txt       | 5 +++++
 2 files changed, 5 insertions(+), 1 deletion(-)
 create mode 100644 requirements.txt

diff --git a/clip_caption_reward.py b/clip_caption_reward.py
index aba1a33..2deb197 100644
--- a/clip_caption_reward.py
+++ b/clip_caption_reward.py
@@ -15,7 +15,6 @@ import sys
 import pathlib
 import json
 from pathlib import Path
-from PIL import Image
 import numpy as np
 import torch
 from torch import nn
diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..ca9f2ee
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,5 @@
+numpy
+torch
+torchvision
+timm
+towhee