Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -258,13 +258,12 @@ def pdb(input_set, mode, impute):
                     st.write('here1')
                     # Save the PDB file to a local file
                     st.write(Path(path_to_output_files / search/'.pdb'))
-                    st.write(response.content)
-                    try:
-                        with open(Path(path_to_output_files / search/'.pdb'), "wb") as f:
-                            st.write('WRITING TO FILE')
-                    except:
-                        st.write('ERROR')
-                    st.write(f"PDB file {search}.pdb downloaded successfully.")
+                    #try:
+                    #    with open(Path(path_to_output_files / search/'.pdb'), "wb") as f:
+                    #        st.write('WRITING TO FILE')
+                    #except:
+                    #    st.write('ERROR')
+                    
 
 
                     from huggingface_hub import HfApi
@@ -274,7 +273,7 @@ def pdb(input_set, mode, impute):
                         path_in_repo="out_files/pdb/pdb_structures/",
                         repo_id="HUBioDataLab/ASCARIS",
                         repo_type="space")
-
+                    st.write(f"PDB file {search}.pdb downloaded successfully.")
                 else:
                     st.write('Here2')
                     st.write(f"Failed to retrieve PDB file for {search}.")