fix repo download separation

app.py

@@ -74,19 +74,8 @@ def toggle_file_input(default_dataset):
     else:
         return gr.update(interactive=True)  # Enable the file input if no default dataset is selected
 
-def download_files(base_path):
-    # clone and cd into UCE repo
+def clone_repo():
     os.system('git clone https://github.com/minwoosun/UCE.git')
-    os.chdir(BASE_PATH)
-    
-    # Add the directory to the Python path
-    sys.path.append(BASE_PATH)
-
-    # Set default dataset path
-    default_dataset_1_path = hf_hub_download(repo_id="minwoosun/uce-misc", filename="100_pbmcs_proc_subset.h5ad")
-    default_dataset_2_path = hf_hub_download(repo_id="minwoosun/uce-misc", filename="1k_pbmcs_proc_subset.h5ad")
-
-    return default_dataset_1_path, default_dataset_2_path 
 
 def main(input_file_path, species, default_dataset, default_dataset_1_path, default_dataset_2_path):
 
@@ -99,9 +88,9 @@ def main(input_file_path, species, default_dataset, default_dataset_1_path, defa
     # # Add the directory to the Python path
     # sys.path.append(BASE_PATH)
 
-    # # Set default dataset path
-    # default_dataset_1_path = hf_hub_download(repo_id="minwoosun/uce-misc", filename="100_pbmcs_proc_subset.h5ad")
-    # default_dataset_2_path = hf_hub_download(repo_id="minwoosun/uce-misc", filename="1k_pbmcs_proc_subset.h5ad")
+    # Set default dataset path
+    default_dataset_1_path = hf_hub_download(repo_id="minwoosun/uce-misc", filename="100_pbmcs_proc_subset.h5ad")
+    default_dataset_2_path = hf_hub_download(repo_id="minwoosun/uce-misc", filename="1k_pbmcs_proc_subset.h5ad")
 
     # If the user selects a default dataset, use that instead of the uploaded file
     if default_dataset == "PBMC 100 cells":
@@ -164,7 +153,7 @@ def main(input_file_path, species, default_dataset, default_dataset_1_path, defa
 if __name__ == "__main__":   
 
     BASE_PATH = '/home/user/app/UCE/'
-    default_dataset_1_path, default_dataset_2_path = download_files(BASE_PATH)
+    clone_repo()
 
     with gr.Blocks() as demo:
         gr.Markdown(
@@ -232,7 +221,7 @@ if __name__ == "__main__":
         # Add the components and link to the function
         run_button.click(
             fn=main, 
-            inputs=[file_input, species_input, default_dataset_input, default_dataset_1_path, default_dataset_2_path], 
+            inputs=[file_input, species_input, default_dataset_input], 
             outputs=[image_output, file_output, pred_output]
         )