Spaces:
Sleeping
Sleeping
app.py
CHANGED
|
@@ -8,7 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_id)
|
|
| 8 |
model = AutoModelForCausalLM.from_pretrained(model_id)
|
| 9 |
|
| 10 |
# Crear pipeline
|
| 11 |
-
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
| 12 |
|
| 13 |
# Funci贸n para generar commit
|
| 14 |
def generar_commit(diff):
|
|
|
|
| 8 |
model = AutoModelForCausalLM.from_pretrained(model_id)
|
| 9 |
|
| 10 |
# Crear pipeline
|
| 11 |
+
pipe = pipeline("text-generation", model=model, tokenizer=tokenizer, return_full_text=False)
|
| 12 |
|
| 13 |
# Funci贸n para generar commit
|
| 14 |
def generar_commit(diff):
|