karthik18AI's picture
max_new_tokens
1979385 verified
raw
history blame
590 Bytes
from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
model_name = "microsoft/codereviewer"
model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
# Example C# code snippet
code_snippet = """
public class HelloWorld {
public static Task Main(string[] args) {
Console.WriteLine("Hello, World!");
}
}
"""
inputs = tokenizer(code_snippet, return_tensors="pt")
outputs = model.generate(**inputs, max_new_tokens=100)
review = tokenizer.decode(outputs[0], skip_special_tokens=True)
print("Code Review:", review)