Update README.md

README.md

@@ -18,7 +18,8 @@ tags:
 # FastApply-7B-v1.0
 
 [Github: kortix-ai/fast-apply](https://github.com/kortix-ai/fast-apply)   
-[Dataset: Kortix/FastApply-dataset-v1.0](https://huggingface.co/datasets/Kortix/FastApply-dataset-v1.0)
+[Dataset: Kortix/FastApply-dataset-v1.0](https://huggingface.co/datasets/Kortix/FastApply-dataset-v1.0)    
+[Try it now on 👉 Google Colab](https://colab.research.google.com/drive/1aBqM8Lqso0Xfgtr75G4LFQivXcChU_36?usp=sharing)
 
 ## Model Details
 
@@ -106,15 +107,15 @@ Provide the complete updated code.<|im_end|>
 input_text = input_text.format(
     original_code=original_code,
     update_snippet=update_snippet,
-).strip() + tokenizer.eos_token 
+).strip() 
 
 # Generate the response
 input_ids = tokenizer.encode(input_text, return_tensors="pt")
-output = model.generate(input_ids, max_length=8192)
-response = tokenizer.decode(output[0])
+output = model.generate(input_ids, max_length=8192,)
+
+response = tokenizer.decode(output[0][len(input_ids[0]):])
+print(response)
 
 # Extract the updated code from the response
 updated_code = response.split("<updated-code>")[1].split("</updated-code>")[0]
-
-print(updated_code)
 ```