Commit
·
a40ed18
1
Parent(s):
e0eeac1
Update app.py
Browse files
app.py
CHANGED
@@ -128,9 +128,9 @@ article = """
|
|
128 |
|
129 |
# British Library Books genre detection demo
|
130 |
|
131 |
-
This demo
|
132 |
-
The model was trained with the [fastai](https://docs.fast.ai/) library on training data drawn from
|
133 |
-
The demo also shows you which parts of the input the model is using most to make its prediction. You can hover over the words to see the
|
134 |
|
135 |
The examples include titles from the BL books collection.
|
136 |
|
@@ -146,12 +146,11 @@ You can find more information about the model [here]((https://doi.org/10.5281/ze
|
|
146 |
|
147 |
## Training data
|
148 |
|
149 |
-
The model is trained on a particular collection of books digitised by the British Library. As a result the model may do less well on titles that look different to this data.
|
150 |
-
In particular the training data, was mostly English, and mostly from the 19th Century.
|
151 |
|
152 |
## Model performance
|
153 |
|
154 |
-
The
|
155 |
|
156 |
|
157 |
```
|
@@ -168,7 +167,6 @@ weighted avg 0.93 0.93 0.93 850
|
|
168 |
### Credits
|
169 |
> This work was partly supported by [Living with Machines](https://livingwithmachines.ac.uk/). This project, funded by the UK Research and Innovation (UKRI) Strategic Priority Fund, is a multidisciplinary collaboration delivered by the Arts and Humanities Research Council (AHRC), with The Alan Turing Institute, the British Library and the Universities of Cambridge, East Anglia, Exeter, and Queen Mary University of London.
|
170 |
> Code for showing attention was adapted from Zach Mueller's (@TheZachMueller) [fastinference](https://muellerzr.github.io/fastinference/) library.
|
171 |
-
|
172 |
"""
|
173 |
|
174 |
gr_interface = gr.Interface(
|
|
|
128 |
|
129 |
# British Library Books genre detection demo
|
130 |
|
131 |
+
This demo allows you to play with a 'genre' detection model which has been trained to predict, from the title of a book, whether it is 'fiction' or 'non-fiction'.
|
132 |
+
The model was trained with the [fastai](https://docs.fast.ai/) library on training data drawn from digitised books at the British Library. These Books are mainly from the 19th Century.
|
133 |
+
The demo also shows you which parts of the input the model is using most to make its prediction. You can hover over the words to see the attention score assigned to that word. This gives you some sense of which words are important to the model in making a prediction.
|
134 |
|
135 |
The examples include titles from the BL books collection.
|
136 |
|
|
|
146 |
|
147 |
## Training data
|
148 |
|
149 |
+
The model is trained on a particular collection of books digitised by the British Library. As a result, the model may do less well on titles that look different to this data. In particular, the training data, was mostly English, and mostly from the 19th Century.
|
|
|
150 |
|
151 |
## Model performance
|
152 |
|
153 |
+
The model's performance on a held-out test set is as follows:
|
154 |
|
155 |
|
156 |
```
|
|
|
167 |
### Credits
|
168 |
> This work was partly supported by [Living with Machines](https://livingwithmachines.ac.uk/). This project, funded by the UK Research and Innovation (UKRI) Strategic Priority Fund, is a multidisciplinary collaboration delivered by the Arts and Humanities Research Council (AHRC), with The Alan Turing Institute, the British Library and the Universities of Cambridge, East Anglia, Exeter, and Queen Mary University of London.
|
169 |
> Code for showing attention was adapted from Zach Mueller's (@TheZachMueller) [fastinference](https://muellerzr.github.io/fastinference/) library.
|
|
|
170 |
"""
|
171 |
|
172 |
gr_interface = gr.Interface(
|