PereLluis13 commited on
Commit
d6222d1
1 Parent(s): 2e567ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
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)