Yapp99 commited on
Commit
e94f76c
·
1 Parent(s): 9e9cb7b

Still trying to fix parser download

Browse files
Files changed (2) hide show
  1. requirements.txt +1 -0
  2. 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 = self.rewriter(diagram).normal_form()
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