Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -468,9 +468,7 @@ def alphafold(input_set, mode, impute):
     ## With new updates, can be updated separately.
 
     pdb_info = pd.read_csv(alphafold_summary, sep='\t')
-    #from huggingface_hub import snapshot_download
-    #alphafold_path = snapshot_download(repo_id="HuBioDataLab/AlphafoldStructures",repo_type = 'dataset')
-    #st.write(alphafold_path)
+
 
     ## Keeping how many models each AlphaFold structure has.
     #model_count = modelCount(alphafold_path)
@@ -625,9 +623,12 @@ def alphafold(input_set, mode, impute):
                             KeyError
                     info_per_model[mod][annot] = annotation_pos_on_pdb_
 
-                
-                pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
-                
+                try:
+                    pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
+                except FileNotFoundError:
+                    pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures3", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
+                except FileNotFoundError:
+                    pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures3", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
 
                 #with gzip.open(pdb_path, mode="rt") as f:
                 #    file_content = f.read()