Update main.py

main.py

@@ -50,13 +50,12 @@ db_retriever = db.as_retriever(search_type="similarity", search_kwargs={"k": 2,
 
 
 # Define prompt template
-prompt_template = """
-As a legal chatbot specializing in the Indian Penal Code (IPC), provide accurate and concise answers based on the context. Respond only if the answer can be derived from the given context; otherwise, reply: "The information is not available in the provided context." Use professional language.
-
+prompt_template = """<s>[INST]As a legal chatbot specializing in the Indian Penal Code, provide a concise and accurate answer based on the given context. Avoid unnecessary details or unrelated content. Only respond if the answer can be derived from the provided context; otherwise, say "The information is not available in the provided context." 
 CONTEXT: {context}
 CHAT HISTORY: {chat_history}
 QUESTION: {question}
 ANSWER:
+</s>[INST]
 """
 prompt = PromptTemplate(template=prompt_template, input_variables=["context", "question", "chat_history"])