Christopher McMaster commited on
Commit
1b37606
1 Parent(s): bc0efd8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -5
app.py CHANGED
@@ -54,11 +54,11 @@ default_text = """# Pancreatitis
54
  """
55
 
56
  title = "Adverse Drug Reaction Highlighting"
57
- description = """This app was made to accompany our recent paper.\n
58
- ADRs will be highlighted in purple, offending medications in green.\n
59
- Hover over a word to see the strength of each prediction on a 0-1 scale.\n
60
- Paper: https://www.medrxiv.org/content/10.1101/2021.12.11.21267504v2\n
61
- Code: https://github.com/AustinMOS/adr-nlp
62
  """
63
 
64
  ner_model = pipeline(task = 'token-classification', model = "austin/adr-ner")
@@ -72,6 +72,7 @@ iface = gr.Interface(_gradio_highlighting,
72
  gr.outputs.HTML(label="ADR Prediction"),
73
  title = title,
74
  description = description,
 
75
  theme = "huggingface"
76
  )
77
  iface.launch()
 
54
  """
55
 
56
  title = "Adverse Drug Reaction Highlighting"
57
+ description = "Named Entity Recognition model to detect ADRs in discharge summaries"
58
+ article = """This app was made to accompany our recent [paper](https://www.medrxiv.org/content/10.1101/2021.12.11.21267504v2).<br>
59
+ ADRs will be highlighted in <p style="color:purple">purple</p>, offending medications in <p style="color:green">green</p>.<br>
60
+ Hover over a word to see the strength of each prediction on a 0-1 scale.<br>
61
+ Our code can be found at [github](https://github.com/AustinMOS/adr-nlp).
62
  """
63
 
64
  ner_model = pipeline(task = 'token-classification', model = "austin/adr-ner")
 
72
  gr.outputs.HTML(label="ADR Prediction"),
73
  title = title,
74
  description = description,
75
+ article = article,
76
  theme = "huggingface"
77
  )
78
  iface.launch()