Updated main

Author: rish-16

gpt2_client/gpt2_client.py

@@ -35,6 +35,7 @@ def __init__(self, model_name='117M', save_dir='models'):
                 
         Methods
         -------
+        download_helper(filename : string)
         load_model(force_download : bool)
         generate(interactive : bool, n_samples : int, words : int, display : bool, return_text: bool) -> list
         generate_batch_from_prompts(prompts : list) -> list