ricklegac commited on
Commit
2fb6558
1 Parent(s): 0093617
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, return_full
13
  # Funci贸n para generar commit
14
  def 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=60, do_sample=True, temperature=0.7)
17
  return [output[0]["generated_text"]]
18
 
19
  # Interfaz Gradio
 
13
  # Funci贸n para generar commit
14
  def 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 Gradio