jk12p/codeexplainer

1import gradio as gr2from transformers import AutoTokenizer, AutoModelForSeq2SeqLM3 4# ✅ Use AutoTokenizer and AutoModel to avoid tokenizer mismatch5tokenizer = AutoTokenizer.from_pretrained("Salesforce/codet5-base")6model = AutoModelForSeq2SeqLM.from_pretrained("Salesforce/codet5-base")7 8# Function to explain code9def explain_code(code_snippet):10    if not code_snippet.strip():11        return "❗ Please enter some code."12    13    input_text = f"summarize: {code_snippet.strip()}"14    input_ids = tokenizer.encode(input_text, return_tensors="pt", truncation=True, max_length=512)15    outputs = model.generate(input_ids, max_length=150, num_beams=4, early_stopping=True)16    explanation = tokenizer.decode(outputs[0], skip_special_tokens=True)17    18    return explanation19 20# Gradio Interface21demo = gr.Interface(22    fn=explain_code,23    inputs=gr.Textbox(lines=15, label="Paste your code here"),24    outputs=gr.Textbox(label="Explanation"),25    title="🧠 Code Explainer using CodeT5",26    description="Paste your code and get a plain English explanation using Salesforce's CodeT5.",27    theme="default"28)29 30demo.launch()31