Update app.py

app.py

@@ -86,7 +86,8 @@ def bot_streaming(message, history):
         # Handle the case where 'image' is not defined at all
         gr.Error("You need to upload an image for LLaVA to work.")
 
-    prompt = f"<|start_header_id|>user<|end_header_id|>\n\n<image>\n{message['text']}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
+    #prompt = f"<|start_header_id|>user<|end_header_id|>\n\n<image>\n{message['text']}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
+    prompt = f"""User:<image>\n{instruction} Falcon:"""
     # print(f"prompt: {prompt}")
     image = Image.open(image)
     inputs = processor(prompt, image, return_tensors='pt').to(0, torch.float16)