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