soltustik commited on
Commit
7942fc5
1 Parent(s): b12de5d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import torch
4
  import torch.nn.functional as F
5
  from transformers import DistilBertTokenizerFast, DistilBertForSequenceClassification
6
 
7
- st.markdown("### Hello, world!")
8
  st.markdown("<img width=200px src='https://media.wired.com/photos/592700e3cfe0d93c474320f1/master/pass/faces-icon.jpg'>", unsafe_allow_html=True)
9
 
10
  model = DistilBertForSequenceClassification.from_pretrained('.')
 
4
  import torch.nn.functional as F
5
  from transformers import DistilBertTokenizerFast, DistilBertForSequenceClassification
6
 
7
+ st.markdown("### Predict tag from title/abstract")
8
  st.markdown("<img width=200px src='https://media.wired.com/photos/592700e3cfe0d93c474320f1/master/pass/faces-icon.jpg'>", unsafe_allow_html=True)
9
 
10
  model = DistilBertForSequenceClassification.from_pretrained('.')