Bug fix: numpy array

app.py

@@ -41,7 +41,7 @@ with torch.no_grad():
         outputs = clip_model.encode_text(inputs)
         all_text_features.append(outputs)
     all_text_features = torch.cat(all_text_features, dim=0)
-    all_text_features = all_text_features.cpu().numpy()
+    all_text_features = all_text_features.cpu()
 
 
 def predict_fn(input_img):

requirements.txt

@@ -4,5 +4,5 @@ ftfy
 regex
 tqdm
 git+https://github.com/openai/CLIP.git
-scikit-learn
+scikit-learn=1.3.0
 scipy