Update app.py

app.py

@@ -55,7 +55,7 @@ tokenizer, model, image_processor, context_len = load_pretrained_model(
 model.config.training = False
 
 # FIXED PROMPT
-FIXED_PROMPT = "What material is this item and how to dispose of it?"
+FIXED_PROMPT = "<image>\nWhat material is this item and how to dispose of it?"
 
 def clear_history():
     state = default_conversation.copy()
@@ -66,9 +66,8 @@ def add_text(state, imagebox, textbox, image_process_mode):
         state = conv_templates[conv_mode].copy()
 
     if imagebox is not None:
-        textbox = DEFAULT_IMAGE_TOKEN + '\n' + FIXED_PROMPT
+        textbox = FIXED_PROMPT
         image = Image.open(imagebox).convert('RGB')
-    if imagebox is not None:
         textbox = (textbox, image, image_process_mode)
     state.append_message(state.roles[0], textbox)
     state.append_message(state.roles[1], None)
@@ -213,7 +212,7 @@ with gr.Blocks(title="CuMo", theme=gr.themes.Default(), css=block_css) as demo:
     
             #cur_dir = os.path.dirname(os.path.abspath(__file__))
             cur_dir = './cumo/serve'
-            default_prompt = "What material is this item and how to dispose of it?"
+            default_prompt = "<image>\nWhat material is this item and how to dispose of it?"
             gr.Examples(examples=[
                 [f"{cur_dir}/examples/0165_CB.jpg", default_prompt],
                 [f"{cur_dir}/examples/0225_PA.jpg", default_prompt],