Spaces:

nielsr
/

imagegpt-completion

Runtime error

nielsr HF Staff commited on Nov 19, 2021

Commit

422ecfe

1 Parent(s): 1df325a

Fix another bug

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,6 +26,7 @@ def process_image(image):
     # create primers
     samples = encoding.pixel_values.numpy()
     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
     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