Update app.py

app.py

@@ -71,8 +71,21 @@ examples = [
  ['World War II', 'followed by'],
  ['Apoorv Umang Saxena', 'country']
 ]
+title = "Interactive demo: KGT5"
+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 (see paper for details). 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.
+<br>
+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.
+For more details see the Github repo or the <a href="https://huggingface.co/apoorvumang/kgt5-base-wikikg90mv2">hf model page</a>.
+"""
+article = "<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>"
+
+
+
 iface = gr.Interface(fn=predict_tail, 
  inputs=[ent_input, rel_input], 
  outputs=output,
+ title=title,
+ description=description,
+ article=article,
  examples=examples,)
 iface.launch()