Apoorv Saxena commited on
Commit
a1b42e5
1 Parent(s): 02f29aa

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +22 -4
app.py CHANGED
@@ -66,10 +66,24 @@ rel_input = gr.inputs.Textbox(lines=1, default="country")
66
  output = gr.outputs.Label()
67
 
68
  examples = [
69
- ['Adrian Kochsiek', 'gender'],
70
- ['Apoorv Umang Saxena', 'family name'],
71
- ['World War II', 'followed by'],
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).
@@ -82,6 +96,10 @@ description = """Demo for <a href='https://arxiv.org/abs/2203.10321'>Sequence-to
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
 
66
  output = gr.outputs.Label()
67
 
68
  examples = [
69
+ ['Adrian Kochsiek', 'sex or gender'],
70
+ ['Apoorv Umang Saxena', 'family name'],
71
+ ['World War II', 'followed by'],
72
+ ['Apoorv Umang Saxena', 'country'],
73
+ ['Ippolito Boccolini', 'writing language'] ,
74
+ ['Roelant', 'writing system'] ,
75
+ ['The Accountant 2227', 'language of work or name'] ,
76
+ ['Microbial Infection and AMR in Hospitalized Patients With Covid 19', 'study type'] ,
77
+ ['Carla Fracci', 'manner of death'] ,
78
+ ['list of programs broadcast by Comet', 'is a list of'] ,
79
+ ['Loreta Podhradí', 'continent'] ,
80
+ ['Opistognathotrema', 'taxon rank'] ,
81
+ ['Museum Arbeitswelt Steyr', 'wheelchair accessibility'] ,
82
+ ['Heliotropium tytoides', 'subject has role'] ,
83
+ ['School bus crash rates on routine and nonroutine routes.', 'sponsor'] ,
84
+ ['Tachigalieae', 'taxon rank'] ,
85
+ ['Irena Salusová', 'place of detention'] ,
86
+
87
  ]
88
  title = "Interactive demo: KGT5"
89
  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).
96
 
97
  article = """
98
  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.
99
+ <br>
100
+ The text representations of the relations and entities can be downloaded from here: <a href="https://storage.googleapis.com/kgt5-wikikg90mv2/rel_alias_list.pickle">https://storage.googleapis.com/kgt5-wikikg90mv2/rel_alias_list.pickle</a> and
101
+ <a href="https://storage.googleapis.com/kgt5-wikikg90mv2/ent_alias_list.pickle">https://storage.googleapis.com/kgt5-wikikg90mv2/ent_alias_list.pickle</a>
102
+ <br>
103
  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>.
104
  """
105