Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -31,7 +31,7 @@ from Bio import SeqIO
 from Bio.PDB import *
 import numpy as np
 
-
+from huggingface_hub import hf_hub_download
 
 
 # FUNCTIONS
@@ -80,8 +80,11 @@ def alphafold(input_set, mode, impute):
     data = clean_data(input_set)
 
     path_to_input_files, path_to_output_files, path_to_domains, fisher_path, path_to_interfaces, alphafold_path, alphafold_summary= manage_files(mode)
-    out_path = path_to_output_files / 'log.txt'
+    #out_path = path_to_output_files / 'log.txt'
     #sys.stdout = open(out_path, 'w')
+    alphafold_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures",repo_type = 'dataset', filename="AF-A0A075B7D0-F1-model_v4.pdb.gz")
+    st.write('aldjflsjflsjfld')
+    st.write(alphafold_path)
     print('Creating directories...')
     file_base = str(Path(alphafold_path / '*'))
     file_str = glob.glob(file_base)[0].split('-')[-1].split('.')[0]