Update

.ipynb_checkpoints/app-checkpoint.py

@@ -102,7 +102,7 @@ def process_pdb(pdb_id, segment):
     with open(prediction_file, "w") as f:
         f.write(result_str)
     
-    return result_str, molecule(pdb_path, random_scores, segment), prediction_file
+    return result_str, molecule(pdb_path, normalized_scores, segment), prediction_file
 
 def molecule(input_pdb, scores=None, segment='A'):
     mol = read_mol(input_pdb)  # Read PDB file content

app.py

@@ -102,7 +102,7 @@ def process_pdb(pdb_id, segment):
     with open(prediction_file, "w") as f:
         f.write(result_str)
     
-    return result_str, molecule(pdb_path, random_scores, segment), prediction_file
+    return result_str, molecule(pdb_path, normalized_scores, segment), prediction_file
 
 def molecule(input_pdb, scores=None, segment='A'):
     mol = read_mol(input_pdb)  # Read PDB file content