fix

app_dialogue.py

@@ -261,12 +261,20 @@ def flag_chat(
 ):
     images = []
     conversations = []
+    prev_ex_is_image = False
     for ex in chat_history:
         if isinstance(ex[0], dict):
             images.append(ex[0]["file"])
-            conversations.append([f"User: <image>"])
+            prev_ex_is_image = True
         else:
-            conversations.append([f"User:{ex[0]}", f"\nAssistant:{ex[1]}"])
+            if prev_ex_is_image:
+                conversations.append([f"User:<image>{ex[0]}", f"\nAssistant:{ex[1]}"])
+            else:
+                conversations.append([f"User:{ex[0]}", f"\nAssistant:{ex[1]}"])
+            prev_ex_is_image = False
+
+    print(f"conversations: {conversations}\n dtype:{type(conversations)}")
+    print(f"chat_history: {chat_history}\n dtype: {type(chat_history)}")
     dope_dataset_writer.flag(
         flag_data=[
             model_selector,
@@ -354,7 +362,6 @@ problematic_dataset_writer = gr.HuggingFaceDatasetSaver(
 #     The second syntax allows inputting an arbitrary number of images.""")
 
 image_fake = gr.Image(visible=False)
-text_fake = gr.Textbox(visible=False)
 
 with gr.Blocks(
     fill_height=True,
@@ -434,7 +441,7 @@ with gr.Blocks(
         [
             model_selector,
             image_fake,
-            text_fake,
+            chatbot,
             decoding_strategy,
             temperature,
             max_new_tokens,
@@ -462,7 +469,7 @@ with gr.Blocks(
         [
             model_selector,
             image_fake,
-            text_fake,
+            chatbot,
             decoding_strategy,
             temperature,
             max_new_tokens,