| import gradio as gr | |
| from transformers import pipeline | |
| # Usamos 'SantaCoder', un modelo de 1.1B par谩metros experto en c贸digo. | |
| # Es mucho m谩s ligero (2GB) y cabe perfectamente en el Space gratuito. | |
| checkpoint = "bigcode/santacoder" | |
| pipe = pipeline("text-generation", model=checkpoint, trust_remote_code=True) | |
| def generate_code(prompt): | |
| # Genera c贸digo basado en tu entrada | |
| output = pipe(prompt, max_length=100, do_sample=True, temperature=0.2) | |
| return output[0]['generated_text'] | |
| # Esta interfaz es lo que evita el "Exit Code: 0" | |
| demo = gr.Interface( | |
| fn=generate_code, | |
| inputs=gr.Textbox(placeholder="Escribe el inicio de una funci贸n Python..."), | |
| outputs="text", | |
| title="Asistente de Programaci贸n IA" | |
| ) | |
| demo.launch() | |