krishnasai99 commited on
Commit
49d2a49
1 Parent(s): e99abbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,6 +56,6 @@ if st.button('Name'):
56
  predicted_ids = torch.argmax(logits, dim=-1)
57
  text = processor.batch_decode(predicted_ids)
58
  str = ''.join(text)
59
- trf = spacy.load('en_core_web_trf')
60
  doc=trf(str)
61
  print(displacy.render(doc,style='ent'))
 
56
  predicted_ids = torch.argmax(logits, dim=-1)
57
  text = processor.batch_decode(predicted_ids)
58
  str = ''.join(text)
59
+ trf = spacy.load('en_core_web_sm')
60
  doc=trf(str)
61
  print(displacy.render(doc,style='ent'))