Upload 3 files

README.md

@@ -9,11 +9,11 @@ app_file: streamlit_app.py
 pinned: false
 ---
 
-# 🧠 Code Explainer (CodeT5-base Summarizer)
+# 🧠 Code Explainer (Final - CodeT5-base)
 
-This app uses `Salesforce/codet5-base-multi-sum`, a model fine-tuned for code summarization.
+This app uses `Salesforce/codet5-base-multi-sum` with optimized prompts for generating docstring-like explanations of Python code.
 
 ## How to Use
-1. Paste Python code in the input box.
+1. Paste your Python code.
 2. Click "Explain".
-3. View the AI-generated explanation of what the function does.
+3. Get a plain English explanation generated by an AI model.

streamlit_app.py

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