removed unnecessary code snippet

run_prothgt_app.py

@@ -151,20 +151,6 @@ def generate_prediction_df(protein_ids, model_paths, model_config_paths, go_cate
 
     heterodata = torch.load(output)
     
-    # Remove unnecessary edge types
-    edge_types_to_remove = [
-        ('Protein', 'protein_function', 'GO_term_F'),
-        ('Protein', 'protein_function', 'GO_term_P'),
-        ('Protein', 'protein_function', 'GO_term_C'),
-        ('GO_term_F', 'rev_protein_function', 'Protein'),
-        ('GO_term_P', 'rev_protein_function', 'Protein'),
-        ('GO_term_C', 'rev_protein_function', 'Protein')
-    ]
-    
-    for edge_type in edge_types_to_remove:
-        if edge_type in heterodata.edge_index_dict:
-            del heterodata.edge_index_dict[edge_type]
-
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 
     for go_cat, model_config_path, model_path in zip(go_category, model_config_paths, model_paths):