Spaces:
Build error
Build error
Still trying to fix parser download
Browse files- requirements.txt +1 -0
- utils.py +11 -11
requirements.txt
CHANGED
|
@@ -11,3 +11,4 @@ qiskit_aer
|
|
| 11 |
spacy
|
| 12 |
streamlit
|
| 13 |
tqdm
|
|
|
|
|
|
| 11 |
spacy
|
| 12 |
streamlit
|
| 13 |
tqdm
|
| 14 |
+
discopy
|
utils.py
CHANGED
|
@@ -30,6 +30,16 @@ MAPPING = {
|
|
| 30 |
ANSATZ = IQPAnsatz(MAPPING, n_layers=1, discard=True)
|
| 31 |
LOGGER = logging.getLogger(__name__)
|
| 32 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 33 |
@dataclass
|
| 34 |
class QNLP_OUTPUT():
|
| 35 |
tokens: str
|
|
@@ -57,16 +67,6 @@ class QNLP():
|
|
| 57 |
'This action only has to happen once.')
|
| 58 |
spacy.cli.download(model)
|
| 59 |
self.nlp = spacy.load(model)
|
| 60 |
-
|
| 61 |
-
self.rewriter = Rewriter([
|
| 62 |
-
'auxiliary',
|
| 63 |
-
'connector',
|
| 64 |
-
'determiner',
|
| 65 |
-
'postadverb',
|
| 66 |
-
'preadverb',
|
| 67 |
-
'prepositional_phrase',
|
| 68 |
-
])
|
| 69 |
-
self.parser = BobcatParser('bert', cache_dir='./')
|
| 70 |
|
| 71 |
def process_sentence(self,
|
| 72 |
input_sentence: str,
|
|
@@ -78,7 +78,7 @@ class QNLP():
|
|
| 78 |
|
| 79 |
def sentence2diagrams(sent: list[str], pb: tqdm = None):
|
| 80 |
diagram = self.parser.sentence2diagram(sent, tokenised=True)
|
| 81 |
-
diagram =
|
| 82 |
|
| 83 |
if pb: pb.update(1)
|
| 84 |
return diagram
|
|
|
|
| 30 |
ANSATZ = IQPAnsatz(MAPPING, n_layers=1, discard=True)
|
| 31 |
LOGGER = logging.getLogger(__name__)
|
| 32 |
|
| 33 |
+
REWRITER = Rewriter([
|
| 34 |
+
'auxiliary',
|
| 35 |
+
'connector',
|
| 36 |
+
'determiner',
|
| 37 |
+
'postadverb',
|
| 38 |
+
'preadverb',
|
| 39 |
+
'prepositional_phrase',
|
| 40 |
+
])
|
| 41 |
+
PARSER = BobcatParser('bert', cache_dir='./')
|
| 42 |
+
|
| 43 |
@dataclass
|
| 44 |
class QNLP_OUTPUT():
|
| 45 |
tokens: str
|
|
|
|
| 67 |
'This action only has to happen once.')
|
| 68 |
spacy.cli.download(model)
|
| 69 |
self.nlp = spacy.load(model)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 70 |
|
| 71 |
def process_sentence(self,
|
| 72 |
input_sentence: str,
|
|
|
|
| 78 |
|
| 79 |
def sentence2diagrams(sent: list[str], pb: tqdm = None):
|
| 80 |
diagram = self.parser.sentence2diagram(sent, tokenised=True)
|
| 81 |
+
diagram = REWRITER(diagram).normal_form()
|
| 82 |
|
| 83 |
if pb: pb.update(1)
|
| 84 |
return diagram
|