Update README.md

README.md

@@ -72,14 +72,28 @@ This is an [OpenLlama model](https://huggingface.co/openlm-research/open_llama_3
 ```python
 from transformers import AutoTokenizer, AutoModelForCausalLM,pipeline
 
-tokenizer = AutoTokenizer.from_pretrained("mwitiderrick/open_llama_3b_instruct_v_0.2")
-model = AutoModelForCausalLM.from_pretrained("mwitiderrick/open_llama_3b_instruct_v_0.2")
-query = "Provide step-by-step instructions for making a sweet chicken bugger"
-text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=500)
+tokenizer = AutoTokenizer.from_pretrained("mwitiderrick/open_llama_3b_code_instruct_v0.1")
+model = AutoModelForCausalLM.from_pretrained("mwitiderrick/open_llama_3b_code_instruct_v0.1")
+query = "write a quick sort algorithm in Python"
+text_gen = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
 output = text_gen(f"### Instruction:\n{query}\n### Response:\n")
 print(output[0]['generated_text'])
 """
+### Instruction:
+write a quick sort algorithm in Python
+### Response:
+def quick_sort(arr):
+    if len(arr) <= 1:
+        return arr
+    else:
+        pivot = arr[len(arr) // 2]
+        left = [x for x in arr if x < pivot]
+        middle = [x for x in arr if x == pivot]
+        right = [x for x in arr if x > pivot]
+        return quick_sort(left) + middle + quick_sort(right)
 
+arr = [1, 2, 3, 4, 5]
+print(quick_sort(arr))
 """
 ```
 ## Metrics