nielsr HF staff commited on
Commit
6358443
1 Parent(s): da92d39

Fix another bug

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -22,7 +22,8 @@ for idx, url in enumerate(urls):
22
 
23
  def process_image(image):
24
  # prepare 8 images, shape (8, 1024)
25
- encoding = feature_extractor([image for _ in range(8)], return_tensors="pt")
 
26
 
27
  # create primers
28
  samples = encoding.pixel_values.numpy()
 
22
 
23
  def process_image(image):
24
  # prepare 8 images, shape (8, 1024)
25
+ batch_size = 8
26
+ encoding = feature_extractor([image for _ in range(batch_size)], return_tensors="pt")
27
 
28
  # create primers
29
  samples = encoding.pixel_values.numpy()