Update app.py

app.py

@@ -232,17 +232,17 @@ def bot(history):
     our_chatbot.conversation.append_message(our_chatbot.conversation.roles[1], None)
     prompt = our_chatbot.conversation.get_prompt()
 
-    # input_ids = (
-    #     tokenizer_image_token(
-    #         prompt, our_chatbot.tokenizer, IMAGE_TOKEN_INDEX, return_tensors="pt"
-    #     )
-    #     .unsqueeze(0)
-    #     .to(our_chatbot.model.device)
-    # )
-    input_ids = tokenizer_image_token(
+    input_ids = (
+        tokenizer_image_token(
             prompt, our_chatbot.tokenizer, IMAGE_TOKEN_INDEX, return_tensors="pt"
-        ).unsqueeze(0).to(our_chatbot.model.device)
-
+        )
+        .unsqueeze(0)
+        .to(our_chatbot.model.device)
+    )
+    # input_ids = tokenizer_image_token(
+    #         prompt, our_chatbot.tokenizer, IMAGE_TOKEN_INDEX, return_tensors="pt"
+    #     ).unsqueeze(0).to(our_chatbot.model.device)
+    print("### input_id",input_ids)
     stop_str = (
         our_chatbot.conversation.sep
         if our_chatbot.conversation.sep_style != SeparatorStyle.TWO
@@ -361,7 +361,7 @@ with gr.Blocks(
             file_types=["image"],
             placeholder="Enter message or upload file...",
             show_label=False,
-            submit_btn=False
+            submit_btn="🚀"
         )
 
         print(cur_dir)