Update README.md

README.md

@@ -16,30 +16,31 @@ license: other
 
 This model was trained using AutoTrain. For more information, please visit [AutoTrain](https://hf.co/docs/autotrain).
 
+Dataset used: codeparrot/xlcost-text-to-code
+
+github: https://github.com/manishzed/LLM-Fine-tune
+
 # Usage
 
 ```python
 
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-model_path = "PATH_TO_THIS_REPO"
-
+model_path = "kr-manish/Mistral-7B-autotrain-text-python-vf1"
 tokenizer = AutoTokenizer.from_pretrained(model_path)
-model = AutoModelForCausalLM.from_pretrained(
-    model_path,
-    device_map="auto",
-    torch_dtype='auto'
-).eval()
-
-# Prompt content: "hi"
-messages = [
-    {"role": "user", "content": "hi"}
-]
-
-input_ids = tokenizer.apply_chat_template(conversation=messages, tokenize=True, add_generation_prompt=True, return_tensors='pt')
-output_ids = model.generate(input_ids.to('cuda'))
-response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
-
-# Model response: "Hello! How can I assist you today?"
-print(response)
+model = AutoModelForCausalLM.from_pretrained(model_path)
+
+#input_text = "Maximum Prefix Sum possible by merging two given arrays | Python3 implementation of the above approach ; Stores the maximum prefix sum of the array A [ ] ; Traverse the array A [ ] ; Stores the maximum prefix sum of the array B [ ] ; Traverse the array B [ ] ;"
+input_text ="Program to convert Centimeters to Pixels | Function to convert centimeters to pixels ; Driver Code"
+# Tokenize input text
+input_ids = tokenizer.encode(input_text, return_tensors="pt")
+
+# Generate output text
+output = model.generate(input_ids, max_length=1024, num_return_sequences=1, do_sample=True)
+
+# Decode and print output
+generated_text = tokenizer.decode(output[0], skip_special_tokens=True)
+print(generated_text)
+
+#Program to convert Centimeters to Pixels | Function to convert centimeters to pixels ; Driver Code [/INST] def cmToPixels ( cm ) : NEW_LINE INDENT return ( ( cm * 100 ) / 17 ) NEW_LINE DEDENT cm = 105.25 NEW_LINE print ( round ( cmToPixels ( cm ) , 3 ) ) NEW_LINE 
 ```