Spaces:

nielsr
/

imagegpt-completion

Runtime error

nielsr HF Staff commited on May 23, 2022

Commit

2c35b0b

1 Parent(s): 714dc8a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ def process_image(image):
     encoding = feature_extractor([image for _ in range(batch_size)], return_tensors="pt")
     # create primers
-    samples = encoding.pixel_values.numpy()
     n_px = feature_extractor.size
     clusters = feature_extractor.clusters
     n_px_crop = 16

     encoding = feature_extractor([image for _ in range(batch_size)], return_tensors="pt")
     # create primers
+    samples = encoding.input_ids.numpy()
     n_px = feature_extractor.size
     clusters = feature_extractor.clusters
     n_px_crop = 16