Update app.py

app.py

@@ -118,7 +118,7 @@ def generate_caption_llava(image_path, caption_bool):
     else:
         text_prompt =f"[INST]  <image>\nCaption this image briefly in objective manner. [/INST]"
     
-    inputs = processor(prompt, Image.open(image_path), return_tensors="pt").to(device)
+    inputs = processor(text_prompt, Image.open(image_path), return_tensors="pt").to(device)
 
     output = model.generate(**inputs, max_new_tokens=100)