Update code/alphafold_featureVector.py

code/alphafold_featureVector.py

@@ -718,10 +718,10 @@ def alphafold(input_set, mode, impute):
                                             st.write('WHY 2')
                                             residue = str(k.split(' ')[1].strip())
                                             if wt == threeToOne(residue):
-                                                sasa = str(k[22:28]).strip('\n')
+                                                sasa = str(k.split(' ')[5]).strip('\n')
                                             elif wt != threeToOne(residue):
                                                 st.write(k)
-                                                sasa = str(k[22:28]).strip('\n') + '*'
+                                                sasa = str(k.split(' ')[5]).strip('\n') + '*'
                                             else:
                                                 sasa = 'nan'