kumar1907 commited on
Commit
33215f4
·
verified ·
1 Parent(s): 72ae69a

Create code_reviewer.py

Browse files
Files changed (1) hide show
  1. code_reviewer.py +9 -0
code_reviewer.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import pipeline
2
+
3
+ # Load model once
4
+ reviewer = pipeline("text2text-generation", model="google/flan-t5-base")
5
+
6
+ def review_code(code_text):
7
+ prompt = f"Review this code and give feedback:\n{code_text}"
8
+ output = reviewer(prompt, max_length=512, do_sample=False)
9
+ return output[0]['generated_text']