Update README.md

README.md

@@ -11,9 +11,10 @@ pipeline_tag: text-generation
 tags:
 - gpt2
 - dpo
+- code
 ---
 
-This model is a finetuned version of ```Sharathhebbar24/chat_gpt2_dpo``` using ```mlabonne/CodeLlama-2-20k```
+This model is a finetuned version of [Sharathhebbar24/code_gpt2_mini_model](https://huggingface.co/Sharathhebbar24/code_gpt2_mini_model) using [Sharathhebbar24/Evol-Instruct-Code-80k-v1](https://huggingface.co/datasets/Sharathhebbar24/Evol-Instruct-Code-80k-v1)
 
 ## Model description
 
@@ -34,7 +35,7 @@ prompt.
 
 ```python
 >>> from transformers import AutoTokenizer, AutoModelForCausalLM
->>> model_name = "Sharathhebbar24/chat_gpt2"
+>>> model_name = "Sharathhebbar24/code_gpt2"
 >>> model = AutoModelForCausalLM.from_pretrained(model_name)
 >>> tokenizer = AutoTokenizer.from_pretrained(model_name)
 >>> def generate_text(prompt):
@@ -42,13 +43,7 @@ prompt.
 >>>  outputs = model.generate(inputs, max_length=64, pad_token_id=tokenizer.eos_token_id)
 >>>  generated = tokenizer.decode(outputs[0], skip_special_tokens=True)
 >>>  return generated[:generated.rfind(".")+1]
->>> prompt = """
->>> user: what are you?
->>> assistant: I am a Chatbot intended to give a python program
->>> user: hmm,  can you write a python program to print Hii Heloo
->>> assistant: Sure Here is a python code.\n print("Hii Heloo")
->>> user: Can you write a Linear search program in python
->>> """
+>>> prompt = "Can you write a Linear search program in Python"
 >>> res = generate_text(prompt)
 >>> res
 ```