Update README.md

README.md

@@ -116,7 +116,7 @@ model =AutoModelForCausalLM.from_pretrained("PipableAI/pip-code-bandit",torch_dt
 tokenizer = tokenizer = AutoTokenizer.from_pretrained("PipableAI/pip-code-bandit")
 
 inputs = tokenizer(prompt, return_tensors="pt").to("cuda")
-outputs = model.generate(**inputs, max_new_tokens=new_tokens,)
+outputs = model.generate(**inputs, max_new_tokens=new_tokens)
 out = (
     tokenizer.decode(outputs[0][inputs.input_ids.shape[-1]:], skip_special_tokens=True)
 )