File size: 752 Bytes
c80c6f7 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 | 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()
|