Updated Prompts

README.md

@@ -38,18 +38,18 @@ Instruction tuned FlanT5-XXL on Legal Clauses data generated via ChatGPT. The mo
 <summary> Click to expand </summary>
 
 ```python
-# pip install accelerate peft bitsandbytes
-import torch
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
-from peft import PeftModel,PeftConfig
-peft_model_id = 'NebulaSense/ContractAssist'
-
-peft_config = PeftConfig.from_pretrained(peft_model_id) 
-
-model =  AutoModelForSeq2SeqLM.from_pretrained(peft_config.base_model_name_or_path, device_map="auto",load_in_8bit=True)
-tokenizer = AutoTokenizer.from_pretrained(peft_config.base_model_name_or_path)
-model = PeftModel.from_pretrained(model, peft_model_id)
-model.eval()
+{
+    "prompt_input": (
+        "Below is an instruction that describes a task, paired with an input that provides further context. "
+        "Write a response that appropriately completes the request.\n\n"
+        "### Instruction:\n{instruction}\n\n### Input:\n{input}\n\n### Response:\n"
+    ),
+    "prompt_no_input": (
+        "Below is an instruction that describes a task. "
+        "Write a response that appropriately completes the request.\n\n"
+        "### Instruction:\n{instruction}\n\n### Response:\n"
+    ),
+} 
 ```
 
 </details>