downloading only the pytorch model and important files, not the other versions of the model

hanlde_form_submit.py

@@ -4,7 +4,7 @@ from time import time
 import streamlit as st
 from grouped_sampling import GroupedSamplingPipeLine
 
-from download_repo import download_repository
+from download_repo import download_pytorch_model
 from prompt_engeneering import rewrite_prompt
 from supported_models import is_supported, SUPPORTED_MODEL_NAME_PAGES_FORMAT, BLACKLISTED_MODEL_NAMES, \
     BLACKLISTED_ORGANIZATIONS
@@ -31,7 +31,7 @@ def create_pipeline(model_name: str, group_size: int) -> GroupedSamplingPipeLine
     if not is_downloaded(model_name):
         download_repository_start_time = time()
         st.write(f"Starts downloading model: {model_name} from the internet.")
-        download_repository(model_name)
+        download_pytorch_model(model_name)
         download_repository_end_time = time()
         download_time = download_repository_end_time - download_repository_start_time
         st.write(f"Finished downloading model: {model_name} from the internet in {download_time:,.2f} seconds.")