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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -172,8 +172,9 @@ def multimodal_rag_query(user_query, num_images_to_retrieve):
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)
 
172
  ]
173
  }
174
  ]
175
+
176
+ #if not getattr(processor, "chat_template", None):
177
+ # processor.chat_template = "{% for message in messages %}{{ message['role'].capitalize() }}: {{ message['content'] }}\n{% endfor %}Assistant:"
178
 
179
  prompt_text = processor.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
180
  inputs = processor(images=retrieved_images_raw, texts=[prompt_text], padding=True, return_tensors="pt").to(device)