ricklegac/commit-bot

1#!/usr/bin/env python32# -*- coding: utf-8 -*-3"""4Created on Thu Apr 17 20:18:35 20255 6@author: rick7"""8 9import gradio as gr10from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline11 12model_id = "mistralai/Mistral-7B-Instruct-v0.1"13 14tokenizer = AutoTokenizer.from_pretrained(model_id)15model = AutoModelForCausalLM.from_pretrained(16    model_id,17    device_map="auto",18    torch_dtype="auto",19)20 21pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)22 23def generar_commit(diff):24    prompt = f"Escribí un mensaje de commit claro, corto y técnico basado en los siguientes cambios:\n{diff}"25    output = pipe(prompt, max_new_tokens=60, do_sample=True, temperature=0.7)26    return [output[0]["generated_text"]]  # <-- esta línea es clave27 28demo = gr.Interface(29    fn=generar_commit,30    inputs=gr.Textbox(lines=10, label="Código diff"),31    outputs="text",32    title="🤖 Commit Generator",33    description="Pega aquí tus cambios (git diff) y generá un mensaje de commit automático.",34)35 36demo.launch()