From 2bdcc837e36a5a20a27fd972f22cc6f31e16ccd3 Mon Sep 17 00:00:00 2001
From: Prithiviraj Damodaran <d.prithiviraj@gmail.com>
Date: Sat, 8 May 2021 12:43:09 +0000
Subject: [PATCH] Update README.md

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index 32a544c..ca69c5c 100644
--- a/README.md
+++ b/README.md
@@ -18,12 +18,12 @@ import warnings
 warnings.filterwarnings("ignore")
 
 
-def set_seed(seed):
+def random_state(seed):
   torch.manual_seed(seed)
   if torch.cuda.is_available():
     torch.cuda.manual_seed_all(seed)
 
-set_seed(1234)
+random_state(1234)
 
 #Init models (make sure you init ONLY once if you integrate this to your code)
 parrot = Parrot(model_tag="prithivida/parrot_paraphraser_on_T5", use_gpu=False)