Update README.md

Add inference code sample

README.md

@@ -24,9 +24,26 @@ based on the information in their previous medical records, current symptoms, ag
 
 ## Intended uses & limitations ⁉️
 
-Sample code snippet:
+Code inference sample:
 ```py
+from peft import PeftModel, PeftConfig
+from transformers import AutoModelForCausalLM, AutoTokenizer
 
+config = PeftConfig.from_pretrained("LaZeAsh/gemma-2b-lahacks")
+model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it")
+model = PeftModel.from_pretrained(model, "LaZeAsh/gemma-2b-lahacks")
+
+tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it")
+
+prompt = "I feel cold I most likely have a "
+
+input_ids = tokenizer.encode(prompt, return_tensors = 'pt')
+
+output = model.generate(input_ids, max_length=50, num_return_sequences=1)
+
+generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
+
+print(generated_text)
 ```
 
 Uses: To use Artificial Intelligence technology to diagnose patient based off of multiple parameters ranging from their age to their