tolist

app.py

@@ -23,7 +23,7 @@ def extract_features(image):
 def update_table(img, df, state, label):
     img_name = state.pop()
     next_img = state[-1]
-    img_embeds = extract_features(img).detach().numpy().squeeze()
+    img_embeds = extract_features(img).detach().numpy().squeeze().tolist()
     print(img_name, img.shape, img_embeds.shape, 'images left:', len(state))
     new_df = pd.DataFrame({'image_name': img_name, 'image_embed': img_embeds, 'label': label}, columns=['image_name', 'image_embed', 'label'])
     print(new_df)