Spaces:
Sleeping
Sleeping
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) |