sagivp commited on
Commit
0030890
·
verified ·
1 Parent(s): 2245413

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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)