Spaces:
Sleeping
Sleeping
Vedant Mahangade
commited on
Commit
•
07067dd
1
Parent(s):
c6e6478
removed unused imports
Browse files
app.py
CHANGED
@@ -1,8 +1,5 @@
|
|
1 |
import streamlit as st
|
2 |
-
import spacy
|
3 |
-
from spacy.tokens import Span
|
4 |
from transformers import pipeline, BigBirdTokenizerFast, AutoModelForTokenClassification
|
5 |
-
# from transformers import PreTrainedTokenizerFast
|
6 |
|
7 |
|
8 |
def pii_app():
|
@@ -38,5 +35,6 @@ def highlight_pii(text, entities):
|
|
38 |
offset += len('<span style="background-color: blue"></span>') # Adjust offset for HTML tags
|
39 |
return highlighted_text
|
40 |
|
|
|
41 |
if __name__ == "__main__":
|
42 |
pii_app()
|
|
|
1 |
import streamlit as st
|
|
|
|
|
2 |
from transformers import pipeline, BigBirdTokenizerFast, AutoModelForTokenClassification
|
|
|
3 |
|
4 |
|
5 |
def pii_app():
|
|
|
35 |
offset += len('<span style="background-color: blue"></span>') # Adjust offset for HTML tags
|
36 |
return highlighted_text
|
37 |
|
38 |
+
|
39 |
if __name__ == "__main__":
|
40 |
pii_app()
|