heymenn commited on
Commit
8a495ab
·
verified ·
1 Parent(s): 75b2af7

Update src/core.py

Browse files
Files changed (1) hide show
  1. src/core.py +5 -4
src/core.py CHANGED
@@ -1,16 +1,17 @@
1
  from src.services.utils import *
2
  from src.services.processor import *
3
 
4
- def process_input(data):
 
 
 
5
  prompt = set_prompt(data.problem)
6
 
7
  constraints = retrieve_constraints(prompt)
8
  constraints_stemmed = stem(constraints, "constraints")
9
 
10
  save_dataframe(constraints_stemmed, "constraints_stemmed.xlsx")
11
-
12
- global_tech, global_tech_embeddings = load_technologies()
13
-
14
  #global_tech, keys, original_tech = preprocess_tech_data(df)
15
 
16
  save_dataframe(global_tech, "global_tech.xlsx")
 
1
  from src.services.utils import *
2
  from src.services.processor import *
3
 
4
+ global_tech, global_tech_embeddings = load_technologies()
5
+
6
+
7
+ def process_input(data, global_tech, global_tech_embeddings):
8
  prompt = set_prompt(data.problem)
9
 
10
  constraints = retrieve_constraints(prompt)
11
  constraints_stemmed = stem(constraints, "constraints")
12
 
13
  save_dataframe(constraints_stemmed, "constraints_stemmed.xlsx")
14
+
 
 
15
  #global_tech, keys, original_tech = preprocess_tech_data(df)
16
 
17
  save_dataframe(global_tech, "global_tech.xlsx")