shreyaspimpalgaonkar
commited on
Update README.md
Browse files
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 |
"""
|