Spaces:

nielsr
/

imagegpt-completion

Runtime error

nielsr HF staff commited on Nov 19, 2021

Commit

ff59753

•

1 Parent(s): 422ecfe

Fix another bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,6 +27,7 @@ def process_image(image):
     # create primers
     samples = encoding.pixel_values.numpy()
     n_px = feature_extractor.size
     n_px_crop = 16
     primers = samples.reshape(-1,n_px*n_px)[:,:n_px_crop*n_px] # crop top n_px_crop rows. These will be the conditioning tokens

     # create primers
     samples = encoding.pixel_values.numpy()
     n_px = feature_extractor.size
+    clusters = feature_extractor.clusters
     n_px_crop = 16
     primers = samples.reshape(-1,n_px*n_px)[:,:n_px_crop*n_px] # crop top n_px_crop rows. These will be the conditioning tokens