shreyaspimpalgaonkar commited on
Commit
d3c3a29
1 Parent(s): 9d16ec0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -29,13 +29,14 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
29
 
30
  def triplextract(model, tokenizer, text, entity_types, predicates):
31
 
32
- input_format = """
 
33
  **Entity Types:**
34
  {entity_types}
35
-
36
  **Predicates:**
37
  {predicates}
38
-
39
  **Text:**
40
  {text}
41
  """
 
29
 
30
  def triplextract(model, tokenizer, text, entity_types, predicates):
31
 
32
+ input_format = """Perform Named Entity Recognition (NER) and extract knowledge graph triplets from the text. NER identifies named entities of given entity types, and triple extraction identifies relationships between entities using specified predicates.
33
+
34
  **Entity Types:**
35
  {entity_types}
36
+
37
  **Predicates:**
38
  {predicates}
39
+
40
  **Text:**
41
  {text}
42
  """