fix outof bounds error

app.py

@@ -22,10 +22,14 @@ def extract_features(image):
 
 def update_table(img, df, state, label):
     img_name = state.pop()
-    next_img = state[-1]
+    next_img = None
+    if len(state) > 0:
+      next_img = state[-1]
     img_embeds = extract_features(img).detach().numpy().squeeze().tolist()
     print(img_name, img.shape, len(img_embeds), 'images left:', len(state))
-    new_df = pd.DataFrame({'image_name': img_name, 'image_embed': img_embeds, 'label': label}, columns=['image_name', 'image_embed', 'label'])
+    new_df = pd.DataFrame({'image_name': img_name, 'image_embed': None, 'label': label}, columns=['image_name', 'image_embed', 'label'], index=[0])
+    print(new_df)
+    new_df.at[0, 'image_embed'] = img_embeds
     print(new_df)
     df = pd.concat([df, new_df])
     return next_img, df, state