aswin-raghavan commited on
Commit
038ac8c
·
1 Parent(s): 82eadf4

still type issue, try values

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -90,8 +90,10 @@ def quantize_embeds(embeds):
90
  return quantized_embeds, closest_bin
91
 
92
  def update_exemplars(df, rng, exemplars, lut):
93
- print(df[['image_embed']], df[['image_embed']].to_numpy())
94
- embeds = df[['image_embed']].to_numpy()
 
 
95
  labels = df[['label']].to_numpy().astype('int')
96
  print(labels)
97
  print(embeds)
 
90
  return quantized_embeds, closest_bin
91
 
92
  def update_exemplars(df, rng, exemplars, lut):
93
+ print(df[['image_embed']], df[['image_embed']].to_numpy().shape)
94
+ print(np.array(df['image_embed'].values.tolist()).shape)
95
+ embeds = np.array(df['image_embed'].values.tolist()) # df[['image_embed']].to_numpy()
96
+ print(df['labels'].to_numpy())
97
  labels = df[['label']].to_numpy().astype('int')
98
  print(labels)
99
  print(embeds)