aeth0r's picture
Update README.md
e80d7c5 verified
metadata
license: mit
base_model:
  - google-t5/t5-small
pipeline_tag: text-generation
tags:
  - literature

compnet-renard/t5-small-literary-relation-extraction

A generative relation extraction model trained on the Despina/project_gutenberg dataset.

Example usage:

from transformers import pipeline

pipeline = pipeline("text2text-generation", model="compnet-renard/t5-small-literary-relation-extraction")
text = "Zarth Arn is Shorr Kann enemy."
pipeline(f"extract relations: {text}")[0]["generated_text"]
>>> "(Zarth Arn, enemy_of, Shorr Kann)"

This model is intended to be used with Renard 0.7+.