update app

app.py

@@ -11,20 +11,20 @@ model_id = "rogerxi/llava-finetune-test"
 pipe = pipeline("image-text-to-text", model=model_id, model_kwargs={"quantization_config": quantization_config})
 
 def update_conversation(new_message, history, image):
-
     if image is None:
         return "Please upload an image first using the widget on the left"
 
-    prompt = "USER: <image>\n"
-
-    for i in range(len(history)):
-        prompt += history[i][0] + '\nASSISTANT: ' + history[i][1] + "\nUSER: "
+    messages = [{"role": "user", "content": "<image>"}]
 
-    prompt = prompt + new_message + '\nASSISTANT: '
+    for user_text, assistant_text in history:
+        messages.append({"role": "user", "content": user_text})
+        messages.append({"role": "assistant", "content": assistant_text})
+    
+    messages.append({"role": "user", "content": new_message})
 
-    outputs = pipe(image=image, text=prompt, generate_kwargs={"max_new_tokens": 200})[0]['generated_text']
+    outputs = pipe(messages=messages, generate_kwargs={"max_new_tokens": 200})[0]['generated_text']
+    return outputs
 
-    return outputs[len(prompt) - 6:]
 
 with gr.Blocks() as demo:
 
@@ -35,4 +35,4 @@ with gr.Blocks() as demo:
           update_conversation, additional_inputs=[image]
       )
 
-demo.launch(share=True, debug=True)
+demo.launch()