Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -10,6 +10,8 @@ model, preprocess_train, preprocess_val = open_clip.create_model_and_transforms(
|
|
10 |
tokenizer = open_clip.get_tokenizer('hf-hub:Marqo/marqo-fashionSigLIP')
|
11 |
|
12 |
def predict(inp):
|
|
|
|
|
13 |
# catgs = [
|
14 |
# "Shirts",
|
15 |
# "SetShirtsPants",
|
@@ -34,7 +36,6 @@ def predict(inp):
|
|
34 |
# "OralCare"
|
35 |
# ]
|
36 |
# text = tokenizer(catgs)
|
37 |
-
# image = preprocess_val(inp).unsqueeze(0)
|
38 |
|
39 |
# with torch.no_grad(), torch.cuda.amp.autocast():
|
40 |
# image_features = model.encode_image(image)
|
|
|
10 |
tokenizer = open_clip.get_tokenizer('hf-hub:Marqo/marqo-fashionSigLIP')
|
11 |
|
12 |
def predict(inp):
|
13 |
+
image = preprocess_val(inp).unsqueeze(0)
|
14 |
+
|
15 |
# catgs = [
|
16 |
# "Shirts",
|
17 |
# "SetShirtsPants",
|
|
|
36 |
# "OralCare"
|
37 |
# ]
|
38 |
# text = tokenizer(catgs)
|
|
|
39 |
|
40 |
# with torch.no_grad(), torch.cuda.amp.autocast():
|
41 |
# image_features = model.encode_image(image)
|