EmanHassan26 commited on
Commit
217f7d4
·
verified ·
1 Parent(s): 068bf0a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)