Christopher McMaster commited on
Commit
1b6eade
1 Parent(s): d12696e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -54,7 +54,12 @@ default_text = """# Pancreatitis
54
  """
55
 
56
  title = "Adverse Drug Reaction Highlighting"
57
- description = "This app was made to accompany our recent paper. ADRs will be highlighted in purple, offending medications in green. Hover over a word to see the strength of each prediction on a 0-1 scale."
 
 
 
 
 
58
 
59
  ner_model = pipeline(task = 'token-classification', model = "austin/adr-ner")
60
  iface = gr.Interface(_gradio_highlighting,
@@ -67,6 +72,6 @@ iface = gr.Interface(_gradio_highlighting,
67
  gr.outputs.HTML(label="ADR Prediction"),
68
  title = title,
69
  description = description,
70
- theme = "darkdefault"
71
  )
72
  iface.launch()
 
54
  """
55
 
56
  title = "Adverse Drug Reaction Highlighting"
57
+ description = """This app was made to accompany our recent paper.
58
+ ADRs will be highlighted in purple, offending medications in green.
59
+ Hover over a word to see the strength of each prediction on a 0-1 scale.
60
+ Paper: https://www.medrxiv.org/content/10.1101/2021.12.11.21267504v2
61
+ Code: https://github.com/AustinMOS/adr-nlp
62
+ """
63
 
64
  ner_model = pipeline(task = 'token-classification', model = "austin/adr-ner")
65
  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()