Update app.py

app.py

@@ -32,7 +32,7 @@ def bot_streaming(message, history):
 
   if image is None:
       gr.Error("You need to upload an image for LLaVA to work.")
-  prompt=f"USER: <image>\n{message['text']}\nASSISTANT:" #f"[INST] <image>\n{message['text']} [/INST]"
+  prompt=f"A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions. USER: <image>\n{message['text']}\nASSISTANT:" #f"[INST] <image>\n{message['text']} [/INST]"
   image = Image.open(image).convert("RGB")
   inputs = processor(prompt, image, return_tensors="pt").to("cuda:0")
 
@@ -43,7 +43,7 @@ def bot_streaming(message, history):
   thread = Thread(target=model.generate, kwargs=generation_kwargs)
   thread.start()
 
-  text_prompt =f"USER:  \n{message['text']}\nASSISTANT: " #f"[INST]  \n{message['text']} [/INST]"
+  text_prompt =f"A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions. USER:  \n{message['text']}\nASSISTANT: " #f"[INST]  \n{message['text']} [/INST]"
   
 
   buffer = ""
@@ -57,7 +57,8 @@ def bot_streaming(message, history):
 
 
 demo = gr.ChatInterface(fn=bot_streaming, title="VLM Playground", examples=[{"text": "What is on the flower?", "files":["./bee.jpg"]},
-                                                                      {"text": "How to make this pastry?", "files":["./baklava.png"]}], 
-                        description="Playground for internal VLMs. Change the model ID and revision under the environments of the Space settings.",
+                                                                      {"text": "How to make this pastry?", "files":["./baklava.png"]},
+                                                                           {"text": "What is this?", "files":["./pizza2.jpeg"]}], 
+                        description="VLM Playground", #for internal VLMs. Change the model ID and revision under the environments of the Space settings.
                         stop_btn="Stop Generation", multimodal=True)
 demo.launch(debug=True)