Upload 2 files

streamlit_app.py

@@ -3,7 +3,7 @@ from model.model_utils import load_model, generate_explanation
 
 st.set_page_config(page_title="Code Explainer", layout="centered")
 
-st.title("🧠 Code Explainer (CodeBERT)")
+st.title("🧠 Code Explainer (CodeT5-small)")
 st.write("Paste your Python code below and get a natural language explanation:")
 
 code_input = st.text_area("Paste Python Code", height=200)
@@ -11,8 +11,8 @@ code_input = st.text_area("Paste Python Code", height=200)
 if st.button("Explain"):
     if code_input.strip():
         with st.spinner("Generating explanation..."):
-            tokenizer, model, device = load_model()
-            explanation = generate_explanation(code_input, tokenizer, model, device)
+            tokenizer, model = load_model()
+            explanation = generate_explanation(code_input, tokenizer, model)
             st.subheader("✅ Explanation")
             st.write(explanation)
     else: