Spaces:
Running
Running
PereLluis13
commited on
Commit
•
d6222d1
1
Parent(s):
2e567ec
Update app.py
Browse files
app.py
CHANGED
@@ -56,6 +56,8 @@ def extract_triplets(text):
|
|
56 |
triplets.append({'head': subject.strip(), 'type': relation.strip(),'tail': object_.strip()})
|
57 |
return triplets
|
58 |
|
|
|
|
|
59 |
tokenizer, model, dataset = load_models()
|
60 |
|
61 |
agree = st.checkbox('Free input', False)
|
|
|
56 |
triplets.append({'head': subject.strip(), 'type': relation.strip(),'tail': object_.strip()})
|
57 |
return triplets
|
58 |
|
59 |
+
st.markdown("""This is a demo for the Findings of EMNLP 2021 paper [REBEL: Relation Extraction By End-to-end Language generation](https://github.com/Babelscape/rebel/blob/main/docs/EMNLP_2021_REBEL__Camera_Ready_.pdf). The pre-trained model is able to extract triplets for up to 200 relation types from Wikidata or be used in downstream Relation Extraction task by fine-tuning. Find the model card [here](https://huggingface.co/Babelscape/rebel-large). Read more about it in the [paper](https://aclanthology.org/2021.findings-emnlp.204) and in the original [repository](https://github.com/Babelscape/rebel).""")
|
60 |
+
|
61 |
tokenizer, model, dataset = load_models()
|
62 |
|
63 |
agree = st.checkbox('Free input', False)
|