Update code_reviewer.py

code_reviewer.py

@@ -1,20 +1,22 @@
 from transformers import pipeline
+import streamlit as st
 
-# Load once
-reviewer = pipeline("text2text-generation", model="google/flan-t5-base")
+# Load once, cache in Streamlit
+@st.cache_resource
+def load_model():
+    return pipeline("text2text-generation", model="google/flan-t5-base")
+
+reviewer = load_model()
 
 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>
     """