Spaces:
Runtime error
Runtime error
Apoorv Saxena
commited on
Commit
·
5fcc4f7
1
Parent(s):
eaa2345
Update app.py
Browse files
app.py
CHANGED
@@ -72,13 +72,18 @@ examples = [
|
|
72 |
['Apoorv Umang Saxena', 'country']
|
73 |
]
|
74 |
title = "Interactive demo: KGT5"
|
75 |
-
description = """Demo for <a href='https://arxiv.org/abs/2203.10321'>Sequence-to-Sequence Knowledge Graph Completion and Question Answering </a> (KGT5). This particular model is T5-base model trained on the task of tail prediction on WikiKG90Mv2 dataset
|
|
|
76 |
<br>
|
77 |
-
Under the hood, this demo concatenates the entity and relation, feeds it to the model and then samples 25 sequences, which are then ranked according to their sequence probabilities.
|
78 |
-
For more details see the Github repo or the <a href="https://huggingface.co/apoorvumang/kgt5-base-wikikg90mv2">hf model page</a>.
|
79 |
"""
|
80 |
-
article = "
|
|
|
|
|
81 |
|
|
|
|
|
|
|
|
|
82 |
|
83 |
|
84 |
iface = gr.Interface(fn=predict_tail,
|
|
|
72 |
['Apoorv Umang Saxena', 'country']
|
73 |
]
|
74 |
title = "Interactive demo: KGT5"
|
75 |
+
description = """Demo for <a href='https://arxiv.org/abs/2203.10321'>Sequence-to-Sequence Knowledge Graph Completion and Question Answering </a> (KGT5). This particular model is a T5-base model trained on the task of tail prediction on WikiKG90Mv2 dataset and obtains 0.239 validation MRR on this task(<a href="https://ogb.stanford.edu/docs/lsc/leaderboards/#wikikg90mv2">leaderboard</a>, see paper for details).
|
76 |
+
To use it, simply give an entity name and relation and click 'submit'. Upto 25 model predictions will show up in a few seconds. The model works best when the exact relation names that it has been trained on are used.
|
77 |
<br>
|
|
|
|
|
78 |
"""
|
79 |
+
#article = """
|
80 |
+
#<p style='text-align: center'><a href='https://arxiv.org/abs/2203.10321'>Sequence-to-Sequence Knowledge Graph Completion and Question Answering </a> | <a href='https://github.com/apoorvumang/kgt5'>Github Repo</a></p>
|
81 |
+
#"""
|
82 |
|
83 |
+
article = """
|
84 |
+
Under the hood, this demo concatenates the entity and relation, feeds it to the model and then samples 25 sequences, which are then ranked according to their sequence probabilities.
|
85 |
+
For more details see the <a href='https://github.com/apoorvumang/kgt5'>Github repo</a> or the <a href="https://huggingface.co/apoorvumang/kgt5-base-wikikg90mv2">hf model page</a>.
|
86 |
+
"""
|
87 |
|
88 |
|
89 |
iface = gr.Interface(fn=predict_tail,
|