varadhbhatnagar
commited on
Commit
•
6d7b7a7
1
Parent(s):
faed0cf
Update README.md
Browse files
README.md
CHANGED
@@ -79,6 +79,10 @@ MRR = 0.3
|
|
79 |
|
80 |
Further details can be found in the paper.
|
81 |
|
|
|
|
|
|
|
|
|
82 |
|
83 |
# Citation
|
84 |
|
@@ -114,9 +118,20 @@ Email: varadhbhatnagar@gmail.com
|
|
114 |
|
115 |
Use the code below to get started with the model.
|
116 |
|
117 |
-
|
118 |
-
|
119 |
-
|
120 |
-
|
121 |
-
|
122 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
79 |
|
80 |
Further details can be found in the paper.
|
81 |
|
82 |
+
# Other Models from same work
|
83 |
+
|
84 |
+
[DBART](https://huggingface.co/varadhbhatnagar/fc-claim-det-DBART)
|
85 |
+
[T5-Base](https://huggingface.co/varadhbhatnagar/fc-claim-det-T5-base)
|
86 |
|
87 |
# Citation
|
88 |
|
|
|
118 |
|
119 |
Use the code below to get started with the model.
|
120 |
|
121 |
+
```
|
122 |
+
from transformers import PegasusForConditionalGeneration, PegasusTokenizerFast
|
123 |
+
tokeizer = PegasusTokenizerFast.from_pretrained('varadhbhatnagar/fc-claim-det-DPEGASUS')
|
124 |
+
model = PegasusForConditionalGeneration.from_pretrained('varadhbhatnagar/fc-claim-det-DPEGASUS')
|
125 |
+
|
126 |
+
text ='world health organisation has taken a complete u turn and said that corona patients neither need isolate nor quarantine nor social distance and it can not even transmit from one patient to another'
|
127 |
+
tokenized_text = tokeizer.encode(text, return_tensors="pt").to(device)
|
128 |
+
|
129 |
+
summary_ids = model.generate(tokenized_text,
|
130 |
+
num_beams=6,
|
131 |
+
no_repeat_ngram_size=2,
|
132 |
+
min_length=5,
|
133 |
+
max_length=15,
|
134 |
+
early_stopping=True)
|
135 |
+
|
136 |
+
output = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
137 |
+
```
|