modification

server.py

@@ -480,7 +480,7 @@ def generate_specific_attack_edge(start_entity, end_entity):
 
     if not torch.cuda.is_available():
         print('We can just set the malicious link equals to the target link, since the generation of malicious link is too slow on cpu')
-        return start_entity, '10', end_entity
+        return entity_to_id[drug_dict[start_entity]], '10', entity_to_id[disease_dict[end_entity]]
     global specific_model
         
     specific_model.to(device)