code-review

app.py

@@ -1,20 +1,20 @@
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-model_name = "Salesforce/codet5-base"
+model_name = "microsoft/codereviewer"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-def generate_summary(code_snippet):
-    inputs = tokenizer(code_snippet, return_tensors="pt", truncation=True, padding="max_length", max_length=512)
-    summary_ids = model.generate(inputs.input_ids, max_length=150, num_beams=4, length_penalty=2.0, early_stopping=True)
-    summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
-    return summary
-
-# Example C# method
+# Example C# code snippet
 code_snippet = """
-public int Add(int a, int b) {
-    return a + b;
+public class HelloWorld {
+    public static Task Main(string[] args) {
+        Console.WriteLine("Hello, World!");
+    }
 }
 """
-summary = generate_summary(code_snippet)
-print("Summary:", summary)
+
+inputs = tokenizer(code_snippet, return_tensors="pt")
+outputs = model.generate(**inputs)
+review = tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+print("Code Review:", review)