Update app.py

app.py

@@ -1,9 +1,9 @@
 import gradio as gr
-from transformers import T5Tokenizer, T5ForConditionalGeneration
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-# Load CodeT5 model and tokenizer
-tokenizer = T5Tokenizer.from_pretrained("Salesforce/codet5-base")
-model = T5ForConditionalGeneration.from_pretrained("Salesforce/codet5-base")
+# ✅ Use AutoTokenizer and AutoModel to avoid tokenizer mismatch
+tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")
+model = AutoModelForSeq2SeqLM.from_pretrained("Salesforce/codet5-base")
 
 # Function to explain code
 def explain_code(code_snippet):
@@ -22,8 +22,8 @@ demo = gr.Interface(
     fn=explain_code,
     inputs=gr.Textbox(lines=15, label="Paste your code here"),
     outputs=gr.Textbox(label="Explanation"),
-    title="🧠 Code Explainer using Hugging Face",
-    description="This tool uses Salesforce's CodeT5 to convert your code into a human-readable explanation. Works on CPU!",
+    title="🧠 Code Explainer using CodeT5",
+    description="Paste your code and get a plain English explanation using Salesforce's CodeT5.",
     theme="default"
 )