Update README.md

README.md

@@ -1,6 +1,9 @@
 ---
-# For reference on model card metadata, see: https://github.com/huggingface/hub-docs/blob/main/modelcard.md?plain=1
-{{card_data}}
+'[object Object]': null
+license: apache-2.0
+language:
+- en
+pipeline_tag: summarization
 ---
 
 # Model Card for Pegasus for Claim Summarization
@@ -30,6 +33,11 @@ This was the best performing model at the time of experimentation.
 - **Repository:** https://github.com/varadhbhatnagar/FC-Claim-Det
 - **Paper:** https://aclanthology.org/2022.coling-1.259/
 
+## Tokenizer
+
+Same as https://huggingface.co/sshleifer/distill-pegasus-cnn-16-4
+
+
 # Uses
 
 <!-- Address questions around how the model is intended to be used, including the foreseeable users of the model and those affected by the model. -->
@@ -60,6 +68,7 @@ those in the corresponding paper as those experiments were conducted in the mont
 <!-- This should link to a Data Card, perhaps with a short stub of information on what the training data is all about as well as documentation related to data pre-processing or additional filtering. -->
 
 [Data](https://github.com/varadhbhatnagar/FC-Claim-Det/blob/main/public_data/released_data.csv)
+
 ## Training Procedure 
 <!-- This relates heavily to the Technical Specifications. Content here should link to that section when it is relevant to the training procedure. -->
 Finetuning the pretrained Distilled PEGASUS model on the 567 pairs released in our paper.
@@ -124,7 +133,7 @@ tokeizer = PegasusTokenizerFast.from_pretrained('varadhbhatnagar/fc-claim-det-DP
 model = PegasusForConditionalGeneration.from_pretrained('varadhbhatnagar/fc-claim-det-DPEGASUS')
 
 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'
-tokenized_text = tokeizer.encode(text, return_tensors="pt").to(device)
+tokenized_text = tokeizer.encode(text, return_tensors="pt")
 
 summary_ids = model.generate(tokenized_text,
                                   num_beams=6,