diff --git a/requirements.txt b/requirements.txt
new file mode 100644
index 0000000..31977f5
--- /dev/null
+++ b/requirements.txt
@@ -0,0 +1,3 @@
+towhee
+torch
+towhee.models
\ No newline at end of file
diff --git a/result1.png b/result1.png
index 2a01eb1..6d574bf 100644
Binary files a/result1.png and b/result1.png differ
diff --git a/result2.png b/result2.png
index d976493..2c14fa4 100644
Binary files a/result2.png and b/result2.png differ
diff --git a/uniformer.py b/uniformer.py
index 28d6a02..523e741 100644
--- a/uniformer.py
+++ b/uniformer.py
@@ -120,9 +120,9 @@ class Uniformer(NNOperator):
         inputs = data.to(self.device)[None, ...]
 
         feats = self.model.forward_features(inputs)
+        feats = feats.flatten(2).mean(-1)
         features = feats.to('cpu').squeeze(0).detach().numpy()
 
-        feats = feats.flatten(2).mean(-1)
         outs = self.model.head(feats)
         post_act = torch.nn.Softmax(dim=1)
         preds = post_act(outs)