Update README.md

README.md

@@ -72,60 +72,62 @@ Any use beyond this is not guaranteed to be accurate.
 ## How to Get Started with the Model
 
 ```
-import torch
 from transformers import pipeline
+import torch
 
-pipe = pipeline("text-generation", 
-                model="dataeaze/dataeaze-RegLLM-zephyr_7b_beta-dzcompli", 
-                torch_dtype=torch.bfloat16, 
+pipe = pipeline("text-generation", model="dataeaze/dataeaze-RegLLM-microsoft_phi_2-dzcompli",
+                torch_dtype=torch.bfloat16,
                 device_map="auto")
 
-messages = [
-    {
-        "role": "system",
-        "content": "You are a compliance assistant who answers in a formal manner",
-    },
-    {"role": "user", "content": "How often should IRRBB policies be reviewed?"},
-]
-prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
-outputs = pipe(prompt, max_new_tokens=120, do_sample=True, temperature=0.1, top_k=50, top_p=0.95)
-print(outputs[0]["generated_text"])
+pipe.tokenizer.pad_token = pipe.tokenizer.eos_token
+
+result = pipe(f"Instruct: Are there specific categories of ECBs with different MAMP requirements?",
+              max_new_tokens=256,
+              do_sample=True,
+              temperature=0.1,
+              top_k=50,
+              top_p=0.95)[0]['generated_text']
+
+print(result)
+
 ```
 
 ## Sample Output
 
 ### Question
-How often should IRRBB policies be reviewed?
+Are there specific categories of ECBs with different MAMP requirements?
 
 ### RegLLM respose
 
 ```
-<|system|>
-You are a compliance assistant who answers in a formal manner</s>
-<|user|>
-How often should IRRBB policies be reviewed?</s>
-<|assistant|>
-At least annually.
-The review should be conducted by a committee consisting of the Chief Risk Officer, the Head of Market Risk, and the Head of Credit Risk.
-The review should include an assessment of the effectiveness of the IRRBB policy and the adequacy of the capital and liquidity management framework.
-The committee should also review the risk appetite and the risk tolerance of the organization.
-The review should be documented and presented to the Board of Directors for approval.
-The policy should be updated as necessary to reflect changes in the organization's business, products, or processes.
+Instruct: Are there specific categories of ECBs with different MAMP requirements?
+Output: Yes, there are various categories of ECBs, each with its own set of MAMP requirements.
+These categories may include infrastructure, commercial real estate, and other specific categories as defined by relevant regulations.
+The specific MAMP requirements for each category may vary, and banks must adhere to these requirements when extending ECBs.
 ```
 
 ### GPT-4 response
-![gpt4 response](./gpt4_reponse.png)
+<table>
+  <tr style="border-spacing: 5px;">
+    <td>
+<img src="chatgpt_ecb_1.png" alt="gpt-4-respnse" width="500" />
+    </td>
+    <td>
+<img src="chatgpt_ecb_2.png" alt="gpt-4-respnse" width="500" />
+    </td>
+  </tr>
+</table>
 
 
 ### Reference 
 
 For evalating truthfulness / hallucination of this response, refer to RBI notification
-[RBI/2022-23/180
-DOR.MRG.REC.102/00-00-009/2022-23](https://rbidocs.rbi.org.in/rdocs/notification/PDFs/NOTI180CF30A8446A704C11BD8267A8D0BB2AC2.PDF) (page 8)
+[RBI/FED/2018-19/67
+FED Master Direction No.5/2018-19](https://rbidocs.rbi.org.in/rdocs/notification/PDFs/5MD2603201979CA1390E9E546869B2A9A92614DEDBF.PDF) (page 9)
 
 Screenshot below
 
-<img src="rbi_reference.png" alt="drawing" width="500"/>
+<img src="rbi_gold_answer_ecb.png" alt="rbi-gold-answer" width="500"/>
 
 
 As you can see, RegLLM has identified the frequency of IRRBB policies, while GPT-4 provides a more general response.