Update README.md

README.md

@@ -20,5 +20,36 @@ model = AutoModelForCausalLM.from_pretrained("unsloth/mistral-7b-bnb-4bit")
 model = PeftModel.from_pretrained(model, "AISimplyExplained/RBI-Notif64")
 tokenizer= AutoTokenizer.from_pretrained("unsloth/mistral-7b-bnb-4bit")
 
+alpaca_prompt = """Below is an instruction. Write a response that appropriately completes the request.
+### Instruction:
+{}
+
+### Response:
+{}"""
+
+
+def formatting_prompts_func(examples):
+
+ inputs = examples["input"]
+ outputs = examples["output"]
+ texts = []
+ for input, output in zip(inputs, outputs):
+ text = alpaca_prompt.format(input, output)
+ texts.append(text)
+ return { "text" : texts, }
+
+
+inputs = tokenizer(
+ [
+ alpaca_prompt.format(
+ f'''What is the reference for the procedure to be followed by RRBs for implementation of Section 51A of UAPA, 1967?
+''',
+ "",
+ )
+ ]*1, return_tensors = "pt").to("cuda")
+
+outputs = model.generate(**inputs, max_new_tokens = 128, use_cache = True)
+output=tokenizer.batch_decode(outputs)[0]
+print(output)
 
 ```