Update app.py

app.py

@@ -87,7 +87,7 @@ def bot_streaming(message, history):
         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"""User:<image>\n{instruction} Falcon:"""
+    prompt = f"""User:<image>\n{message} Falcon:"""
     # print(f"prompt: {prompt}")
     image = Image.open(image)
     inputs = processor(prompt, image, return_tensors='pt').to(0, torch.float16)