Bug fix: numpy array
Browse files- app.py +1 -1
- requirements.txt +1 -1
app.py
CHANGED
@@ -41,7 +41,7 @@ with torch.no_grad():
|
|
41 |
outputs = clip_model.encode_text(inputs)
|
42 |
all_text_features.append(outputs)
|
43 |
all_text_features = torch.cat(all_text_features, dim=0)
|
44 |
-
all_text_features = all_text_features.cpu()
|
45 |
|
46 |
|
47 |
def predict_fn(input_img):
|
|
|
41 |
outputs = clip_model.encode_text(inputs)
|
42 |
all_text_features.append(outputs)
|
43 |
all_text_features = torch.cat(all_text_features, dim=0)
|
44 |
+
all_text_features = all_text_features.cpu()
|
45 |
|
46 |
|
47 |
def predict_fn(input_img):
|
requirements.txt
CHANGED
@@ -4,5 +4,5 @@ ftfy
|
|
4 |
regex
|
5 |
tqdm
|
6 |
git+https://github.com/openai/CLIP.git
|
7 |
-
scikit-learn
|
8 |
scipy
|
|
|
4 |
regex
|
5 |
tqdm
|
6 |
git+https://github.com/openai/CLIP.git
|
7 |
+
scikit-learn=1.3.0
|
8 |
scipy
|