Girgis711 commited on
Commit
83301e6
1 Parent(s): 4008bad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -1,9 +1,14 @@
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
- pipe = pipeline('sentiment-analysis')
 
 
 
 
5
  text = st.text_area('enter some text:')
6
 
7
  if text:
8
- out = pipe(text)
 
9
  st.json(out)
 
1
  import streamlit as st
2
  from transformers import pipeline
3
 
4
+ from span_marker import SpanMarkerModel
5
+
6
+ # Download from the 🤗 Hub
7
+ model = SpanMarkerModel.from_pretrained("tomaarsen/span-marker-bert-base-acronyms")
8
+ # Run inference
9
  text = st.text_area('enter some text:')
10
 
11
  if text:
12
+ out = model.predict(text)
13
+
14
  st.json(out)