ricklegac commited on
Commit
b499a85
·
1 Parent(s): f1a4d52
Files changed (1) hide show
  1. commit.py +1 -1
commit.py CHANGED
@@ -23,7 +23,7 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
23
  def 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"]
27
 
28
  demo = gr.Interface(
29
  fn=generar_commit,
 
23
  def 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 clave
27
 
28
  demo = gr.Interface(
29
  fn=generar_commit,