fine-tune

app.py

@@ -4,10 +4,13 @@ model_name = "microsoft/codereviewer"
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 
-# Example C# code snippet
+# Example C# code snippet with more context
 code_snippet = """
+// This class demonstrates a simple HelloWorld program
 public class HelloWorld {
+    // Main method
     public static Task Main(string[] args) {
+        // Print Hello, World! to the console
         Console.WriteLine("Hello, World!");
     }
 }
@@ -17,4 +20,4 @@ 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)
+print("Code Review:", review)