ricklegac commited on
Commit
1f2d1eb
1 Parent(s): b37778a
Files changed (1) hide show
  1. app.py +1 -1
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):