Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -30,8 +30,9 @@ from alphafold_model import *
 def getModelInfo(uniprotID, varPos, wt, models_we_need, path_to_output_files ):
     modelInfo = {}
     st.write(uniprotID)
-    st.write(f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz")
+    
     for mod in models_we_need:
+        st.write(f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz")
         try:
             pdb_path = hf_hub_download(repo_id="HuBioDataLab/AlphafoldStructures", filename=f"AF-{uniprotID}-F{mod}-model_v4.pdb.gz",repo_type = 'dataset')
         except u.EntryNotFoundError: