Update app.py

app.py

@@ -15,7 +15,7 @@ MODEL_NAME = MODEL_ID.split("/")[-1]
 
 TITLE = "<h1>VL-Chatbox</h1>"
 
-DESCRIPTION = f'<p>A SPACE FOR VLM MODELS</p><br><h3><center>MODEL NOW: <a href="https://hf.co/{MODEL_ID}">{MODEL_NAME}</a></center></h3>'
+DESCRIPTION = f'<center><p>A SPACE FOR VLM MODELS</p><br><h3>MODEL NOW: <a href="https://hf.co/{MODEL_ID}">{MODEL_NAME}</a></center></h3>'
 
 CSS = """
 .duplicate-button {
@@ -28,9 +28,6 @@ h1 {
     text-align: center;
     display: block;
 }
-p {
-    text-align: center;
-}
 """
 
 model = AutoModelForCausalLM.from_pretrained(
@@ -57,9 +54,10 @@ def stream_chat(message, history: list, temperature: float, max_length: int, top
             image = None
             conversation.append({"role": "user", "content": message['text']})
         else:
-            image = Image.open(history[0][0][0])
+            #image = Image.open(history[0][0][0])
             for prompt, answer in history:
                 if answer is None:
+                    image = Image.open(prompt[0])
                     conversation.extend([{"role": "user", "content": ""},{"role": "assistant", "content": ""}])
                 else:
                     conversation.extend([{"role": "user", "content": prompt}, {"role": "assistant", "content": answer}])
@@ -96,7 +94,7 @@ chatbot = gr.Chatbot(height=450)
 chat_input = gr.MultimodalTextbox(
     interactive=True,
     file_types=["image"],
-    placeholder="Enter message or upload file...",
+    placeholder="Enter message or upload a file one time...",
     show_label=False,
 
 )