MarkelFe commited on
Commit
4ab7d61
·
1 Parent(s): 61915c1

Remove forgotten print

Browse files
Files changed (1) hide show
  1. app.py +0 -1
app.py CHANGED
@@ -28,7 +28,6 @@ def sortu_testua(alderdia, testua, max_tokens, conf, ngram, beams, top_k, top_p)
28
  tokens = tokenizer(prompt, return_tensors="pt").to(device)
29
  generation = model.generate(inputs=tokens['input_ids'], attention_mask = tokens['attention_mask'], **options)[0]
30
  text = tokenizer.decode(generation)
31
- print(re.split("\[(.*?)\] ", text))
32
  return re.split("\[(.*?)\] ", text)[-1]
33
 
34
  def sortu_testu_guztiak(testua, max_tokens, conf, ngram, beams, top_k, top_p):
 
28
  tokens = tokenizer(prompt, return_tensors="pt").to(device)
29
  generation = model.generate(inputs=tokens['input_ids'], attention_mask = tokens['attention_mask'], **options)[0]
30
  text = tokenizer.decode(generation)
 
31
  return re.split("\[(.*?)\] ", text)[-1]
32
 
33
  def sortu_testu_guztiak(testua, max_tokens, conf, ngram, beams, top_k, top_p):