stopped downloading the models when the servers start

app.py

@@ -9,9 +9,7 @@ from torch.cuda import CudaError
 
 from available_models import AVAILABLE_MODELS
 from hanlde_form_submit import on_form_submit
-from on_server_start import main as on_server_start_main
 
-# on_server_start_main()
 
 st.title("Grouped Sampling Demo")
 

on_server_start.py

@@ -1,37 +0,0 @@
-"""
-A script that is run when the server starts.
-"""
-from concurrent.futures import ThreadPoolExecutor
-
-from transformers import logging as transformers_logging
-from huggingface_hub import logging as huggingface_hub_logging
-
-from available_models import AVAILABLE_MODELS
-from download_repo import download_pytorch_model
-
-
-def disable_progress_bar():
-    """
-    Disables the progress bar when downloading models.
-    """
-    transformers_logging.disable_progress_bar()
-    huggingface_hub_logging.disable_propagation()
-
-
-def download_useful_models():
-    """
-    Downloads the models that are useful for this project.
-    So that the user doesn't have to wait for the models to download when they first use the app.
-    """
-    print("Downloading useful models. It might take a while...")
-    with ThreadPoolExecutor() as executor:
-        executor.map(download_pytorch_model, AVAILABLE_MODELS)
-
-
-def main():
-    # disable_progress_bar()
-    download_useful_models()
-
-
-if __name__ == "__main__":
-    main()

tests.py

@@ -3,18 +3,11 @@ import os
 import pytest as pytest
 from grouped_sampling import GroupedSamplingPipeLine, get_full_models_list, UnsupportedModelNameException
 
-from on_server_start import download_useful_models
 from hanlde_form_submit import create_pipeline, on_form_submit
 
 HUGGING_FACE_CACHE_DIR = "/home/yoni/.cache/huggingface/hub"
 
 
-def test_on_server_start():
-    download_useful_models()
-    assert os.path.exists(HUGGING_FACE_CACHE_DIR)
-    assert len(os.listdir(HUGGING_FACE_CACHE_DIR)) > 0
-
-
 def test_on_form_submit():
     model_name = "gpt2"
     output_length = 10