Update README.md

README.md

@@ -21,6 +21,7 @@ from tokenizers import Tokenizer, models, pre_tokenizers, trainers, Regex
 import tokenizers
 from tokenizers import Tokenizer, models, decoders, processors
 from tokenizers import pre_tokenizers, trainers, Regex
+import huggingface_hub
 
 class Attention(tf.keras.layers.Layer):
     def __init__(self,  
@@ -51,7 +52,9 @@ class Attention(tf.keras.layers.Layer):
         return config
 
 #download model
-model = tf.keras.models.load_model(<path_to_model>,
+model = tf.keras.models.load_model(huggingface_hub.hf_hub_download('vkovenko/deep_lstm_attention_ukr_reviews_rating_estimation',
+                                'deep_lstm_attention_w2v_huber.h5',
+                               local_dir='model'),
                                   compile=False,
                                   custom_objects={'Attention':Attention})
 
@@ -101,9 +104,14 @@ class BPETokenizer:
         else:
             return decoded
 #download tokenizer
-tokenizer = BPETokenizer(vocab=<path to vocab>,
-            merges=<path to merges>
+tokenizer = BPETokenizer(vocab=huggingface_hub.hf_hub_download('vkovenko/deep_lstm_attention_ukr_reviews_rating_estimation',
+                                'tokenizer_30k.json',
+                               local_dir='model'),
+            merges=huggingface_hub.hf_hub_download('vkovenko/deep_lstm_attention_ukr_reviews_rating_estimation',
+                                'merges_tokenizer.txt',
+                               local_dir='model')
                         )
+
 ```