varadhbhatnagar commited on
Commit
6d7b7a7
1 Parent(s): faed0cf

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +21 -6
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
- <details>
118
- <summary> Click to expand </summary>
119
-
120
- {{ get_started_code | default("[More Information Needed]", true)}}
121
-
122
- </details>
 
 
 
 
 
 
 
 
 
 
 
 
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
+ ```