ricklegac/commit-bot

1import gradio as gr2from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline3 4model_id = "microsoft/phi-1_5"  # modelo 100% libre5 6# Carga del modelo sin device_map ni dtype7tokenizer = AutoTokenizer.from_pretrained(model_id)8model = AutoModelForCausalLM.from_pretrained(model_id)9 10# Crear pipeline11pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, return_full_text=False)12 13# Función para generar commit14def generar_commit(diff):15    prompt = f"Escribí solo el mensaje de commit para el siguiente cambio de código:\n{diff}"16    output = pipe(prompt, max_new_tokens=40, do_sample=True, temperature=0.7)17    return [output[0]["generated_text"]]18 19# Interfaz Gradio20demo = gr.Interface(21    fn=generar_commit,22    inputs=gr.Textbox(lines=10, label="Código diff"),23    outputs="text",24    title="🤖 Commit Generator",25    description="Pega aquí tus cambios (git diff) y generá un mensaje de commit automático.",26)27 28demo.launch()