nazlikacmaz/code-explainer
0
1import gradio as gr2from transformers import pipeline3 4generator = pipeline("text-generation", model="distilgpt2")5 6def run_prompt(prompt):7 result = generator(prompt, max_new_tokens=100, do_sample=False)8 return result[0]["generated_text"]9 10iface = gr.Interface(11 fn=run_prompt,12 inputs=gr.Textbox(label="Prompt", lines=5),13 outputs="text",14 title="Code Explainer (Fast)",15 description="Uses distilgpt2 for ultra-fast explanation"16)17 18iface.launch() 