Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -238,7 +238,7 @@ def pdb(input_set, mode, impute):
             pdb_data = response.text
             pdb_parser = PDBParser(QUIET=True)  # QUIET=True suppresses warnings
             pdb_file_content = StringIO(pdb_data)
-            structure = pdb_parser.get_structure(pdb_code, pdb_file_content)
+            structure = pdb_parser.get_structure(search, pdb_file_content)
             st.write(structure)
             ppb = PPBuilder()
             for model in structure: