Spaces:
Sleeping
Sleeping
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
|