Update code/pdb_featureVector.py

code/pdb_featureVector.py

@@ -220,13 +220,12 @@ def pdb(input_set, mode, impute):
             st.write('response', response)
             response.raise_for_status()  # Check for a successful response
             data = response.json()
-            st.write(list(list(list(data.values())[0].values())[0].keys()))
-            return list(list(list(data.values())[0].values())[0].keys())
-        
+            
+            return list(list(list(data.values())[0].values())[0].keys()))
         for search in pdbs:
             # Step 1: Fetch the PDB file
             pdb_url = f"https://files.rcsb.org/download/{search}.pdb"
-            st.write(pdb_url)
+            st.write
             try:
                 response = requests.get(pdb_url)
                 response.raise_for_status()  # Check for a successful response