Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -172,6 +172,8 @@ def multimodal_rag_query(user_query, num_images_to_retrieve):
|
|
| 172 |
]
|
| 173 |
}
|
| 174 |
]
|
|
|
|
|
|
|
| 175 |
|
| 176 |
prompt_text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
| 177 |
inputs = processor(images=retrieved_images_raw, texts=[prompt_text], padding=True, return_tensors="pt").to(device)
|
|
|
|
| 172 |
]
|
| 173 |
}
|
| 174 |
]
|
| 175 |
+
if not getattr(processor, "chat_template", None):
|
| 176 |
+
processor.chat_template = "{% for message in messages %}{{ message['role'].capitalize() }}: {{ message['content'] }}\n{% endfor %}Assistant:"
|
| 177 |
|
| 178 |
prompt_text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
|
| 179 |
inputs = processor(images=retrieved_images_raw, texts=[prompt_text], padding=True, return_tensors="pt").to(device)
|