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"[INST] <image>\n{message['text']} [/INST]"
+  prompt=f"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"[INST]  \n{message['text']} [/INST]"
+  text_prompt =f"USER: \n{message['text']}\nASSISTANT:" #f"[INST]  \n{message['text']} [/INST]"
   
 
   buffer = ""