radames commited on
Commit
1ac1ce8
1 Parent(s): a607b23
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -26,7 +26,7 @@ class PreTrainedPipeline():
26
  - "feature_vector": A list of floats corresponding to the image embedding.
27
  """
28
 
29
- batch_dict = self.tokenizer(inputs, max_length=512,
30
  padding=True, truncation=True, return_tensors='pt')
31
  with torch.no_grad():
32
  outputs = self.model(**batch_dict)
 
26
  - "feature_vector": A list of floats corresponding to the image embedding.
27
  """
28
 
29
+ batch_dict = self.tokenizer([inputs], max_length=512,
30
  padding=True, truncation=True, return_tensors='pt')
31
  with torch.no_grad():
32
  outputs = self.model(**batch_dict)