Update code_reviewer.py

code_reviewer.py

@@ -1,22 +1,20 @@
 from transformers import pipeline
-import streamlit as st
 
-# Load once, cache in Streamlit
-@st.cache_resource
-def load_model():
-    return pipeline("text2text-generation", model="google/flan-t5-base")
-
-reviewer = load_model()
+# Load once
+reviewer = pipeline("text2text-generation", model="google/flan-t5-base")
 
 def review_code(code_text):
+    # Ask model for both feedback + improved code
     prompt = f"""
     You are a code review assistant. Analyze the following code diff or snippet:
     {code_text}
+
     1. Give clear review feedback with improvements.
     2. Then provide a refactored/improved version of the code.
     Format the response as:
     FEEDBACK:
     <your feedback here>
+
     CODE:
     <improved code here>
     """