image embedding to the right

app.py

@@ -27,7 +27,7 @@ def update_table(img, df, state, label):
       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': None, 'label': label}, columns=['image_name', 'image_embed', 'label'], index=[0])
+    new_df = pd.DataFrame({'image_name': img_name, 'label': label, 'image_embed': None}, columns=['image_name', 'image_embed', 'label'], index=[0])
     print(new_df)
     new_df.at[0, 'image_embed'] = img_embeds
     print(new_df)
@@ -46,10 +46,10 @@ with gr.Blocks(title="End-User Personalization") as demo:
     # start_button = gr.Button(label="Start")
     image_display = gr.Image()
     # text_display = gr.Text()
-    with gr.Column():
-        upvote = gr.Button("Up")
-        downvote = gr.Button("down")
-    annotated_samples = gr.Dataframe(headers=['image_name', 'image_embed', 'label'], row_count=(1, 'dynamic'), 
+    with gr.Row():
+        upvote = gr.Button("👍")
+        downvote = gr.Button("👎")
+    annotated_samples = gr.Dataframe(headers=['image_name', 'label', 'image_embed'], row_count=(1, 'dynamic'), 
                                      col_count=(3, 'fixed'), label='Annotations', wrap=True)
     upvote.click(update_table_up, inputs=[image_display, annotated_samples, train_images], outputs=[image_display, annotated_samples, train_images])
     downvote.click(update_table_down, inputs=[image_display, annotated_samples, train_images], outputs=[image_display,annotated_samples, train_images])