add requirements

Signed-off-by: gexy5 <xinyu.ge@zilliz.com>

requirements.txt

@@ -0,0 +1,3 @@
+towhee
+torch
+towhee.models

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)