Spaces:
Sleeping
Sleeping
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=
|
| 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
|