Sasiraj01 commited on
Commit
81c2bbd
·
verified ·
1 Parent(s): 8fc23d7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -29,7 +29,7 @@ def multimodal_rag(image, question):
29
  context = query_engine.query(question)
30
 
31
  # Step 2: Process with LLaVA
32
- prompt = f"Context: {context}
33
 
34
  Question: {question}"
35
  inputs = processor(prompt, image, return_tensors="pt").to(model.device)
 
29
  context = query_engine.query(question)
30
 
31
  # Step 2: Process with LLaVA
32
+ prompt = f"Context: {context}\n\nQuestion: {question}"
33
 
34
  Question: {question}"
35
  inputs = processor(prompt, image, return_tensors="pt").to(model.device)