oroszgy commited on
Commit
5db52bb
1 Parent(s): 238484d

feat: reducing the demo loading by using the common NLP object

Browse files
Files changed (1) hide show
  1. examples/relation.py +11 -20
examples/relation.py CHANGED
@@ -1,15 +1,12 @@
1
  import gradio as gr
2
-
3
- import spacy
4
  import pandas as pd
5
- from spacy import displacy
6
 
 
7
  from resources import triples
8
 
9
- nlp = spacy.load("hu_core_news_trf")
10
 
11
  def process(text: str) -> pd.DataFrame:
12
- doc = nlp(text)
13
  tuples_to_list = list()
14
 
15
  tuples = triples.subject_verb_object_triples(doc)
@@ -33,25 +30,19 @@ def process(text: str) -> pd.DataFrame:
33
  object += str(obj_multiple) + ", "
34
  object = object[:-2]
35
 
36
- relation_list = [[subject,
37
- verb,
38
- object]]
39
 
40
  return pd.DataFrame(relation_list, columns=['Subject', 'Verb', 'Object'])
41
 
42
- EXAMPLES = ["Anna éppen most házat épít magának.",
43
- "Noémi gulyáslevest szeret főzni, ha éhes.",
44
- "Balázs Jéghideg helyi ananászlevet ivott Hawaii fehér homokos partján.",
45
- "Júliska fagyit árul a nyáron teljes állásban.",
46
- "Einstein megmutatta a házát építés közben.",
47
- "Hawking nyilatkozott egy levelet, miszerint a felfedezései az élete legizgalmasabb eseményei voltak."]
48
-
49
- # process(EXAMPLES[0])
50
 
51
- # displacy checker
52
- # text = nlp(EXAMPLES[1])
53
- # displacy.serve(text, style="dep")
54
- # alma = 41
 
 
 
 
55
 
56
  demo = gr.Interface(
57
  fn=process,
 
1
  import gradio as gr
 
 
2
  import pandas as pd
 
3
 
4
+ from examples.common import NLP
5
  from resources import triples
6
 
 
7
 
8
  def process(text: str) -> pd.DataFrame:
9
+ doc = NLP(text)
10
  tuples_to_list = list()
11
 
12
  tuples = triples.subject_verb_object_triples(doc)
 
30
  object += str(obj_multiple) + ", "
31
  object = object[:-2]
32
 
33
+ relation_list = [[subject, verb, object]]
 
 
34
 
35
  return pd.DataFrame(relation_list, columns=['Subject', 'Verb', 'Object'])
36
 
 
 
 
 
 
 
 
 
37
 
38
+ EXAMPLES = [
39
+ "Anna éppen most házat épít magának.",
40
+ "Noémi gulyáslevest szeret főzni, ha éhes.",
41
+ "Balázs jéghideg helyi ananászlevet ivott Hawaii fehér homokos partján.",
42
+ "Júliska fagyit árul a nyáron teljes állásban.",
43
+ "Einstein megmutatta a házát építés közben.",
44
+ "Hawking nyilatkozott egy levelet, miszerint a felfedezései az élete legizgalmasabb eseményei voltak."
45
+ ]
46
 
47
  demo = gr.Interface(
48
  fn=process,