Update README.md

README.md

@@ -491,6 +491,8 @@ print(outputs[0].outputs[0].text)
 
 This model can also generate questions and answers based on a piece of text. This can be useful for pre-indexing a database or fine-tuning IR models that will then be used for RAG.
 
+To define the language of the Q&A, you can optionally include a `<<Q&A Language>>` control code after the text
+
 <details>
   <summary>Prompt style</summary>
 
@@ -499,6 +501,9 @@ This model can also generate questions and answers based on a piece of text. Thi
 <<Q&A Generation Context>>
 Japan's current account surplus in July was 3.2 trillion yen, the highest monthly surplus on record for the month of July.
 However, the surplus continues to be driven by the primary income balance, which recorded a surplus of 4.4 trillion yen in July, the highest monthly figure on record.
+
+<<Q&A Language>>
+English
 ```
 
 ### Output: 
@@ -524,11 +529,14 @@ sampling_params = SamplingParams(temperature=0.2, top_p=0.95, max_tokens=128)
 
 context = "Finance Minister Shunichi Suzuki appointed Kenji Suwazono, former Director-General of the Customs and Tariff Bureau at the Ministry of Finance, as the new Executive Director of the Bank of Japan effective the 10th. Suwazono succeeds Masaaki Kaizuka, whose term ended on the 9th, and his term will last for four years.",
 
-def create_qagen_prompt(context):
+def create_qagen_prompt(context, language_name = None):
 
     str_inputs = f"""<<Q&A Generation Context>>
 {context}"""
 
+    if language_name is not None:
+        str_inputs += "\n\n<<Q&A Language>>\n{language_name}"
+
     chat = [
       {"role": "user", "content": str_inputs},
     ]