GiordanoB commited on
Commit
3bea5e2
1 Parent(s): 909aa44

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +26 -2
README.md CHANGED
@@ -24,8 +24,32 @@ It achieves the following results on the evaluation set:
24
 
25
  ## Model description
26
 
27
- More information needed
28
-
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
29
  ## Intended uses & limitations
30
 
31
  More information needed
 
24
 
25
  ## Model description
26
 
27
+ for i in range(0,14):
28
+ input_ids = tokenizer(
29
+ [WHITESPACE_HANDLER(sumariosDuplos[i])],
30
+ return_tensors="pt",
31
+ padding="max_length",
32
+ truncation=True,
33
+ max_length=512
34
+ )["input_ids"]
35
+
36
+ output_ids = model.generate(
37
+ input_ids=input_ids,
38
+ max_length=200,
39
+ min_length=75,
40
+ no_repeat_ngram_size=2,
41
+ num_beams=5
42
+ )[0]
43
+
44
+ summary = tokenizer.decode(
45
+ output_ids,
46
+ skip_special_tokens=True,
47
+ clean_up_tokenization_spaces=False
48
+ )
49
+
50
+ sumariosFinal.append(summary)
51
+ print(i,"\n",summary,"\n")
52
+
53
  ## Intended uses & limitations
54
 
55
  More information needed