imamnurby commited on
Commit
5f2f1d6
·
1 Parent(s): d4ebedd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -9
app.py CHANGED
@@ -13,12 +13,13 @@ components = initialize_all_components(config)
13
  db_metadata = components[0]
14
  db_constructor = components[1]
15
  db_params = components[2]
16
- model_retrieval = components[3]
17
- model_generative = components[4]
18
- tokenizer_generative = components[5]
19
- model_classifier = components[6]
20
- classifier_head = components[7]
21
- tokenizer_classifier = components[8]
 
22
 
23
  def call_predict_api(
24
  input_query,
@@ -26,7 +27,7 @@ def call_predict_api(
26
  model_generative,
27
  model_classifier, classifier_head,
28
  tokenizer_generative, tokenizer_classifier,
29
- db_metadata, db_constructor, db_params,
30
  config
31
  ):
32
  '''
@@ -38,7 +39,7 @@ def call_predict_api(
38
  model_generative,
39
  model_classifier, classifier_head,
40
  tokenizer_generative, tokenizer_classifier,
41
- db_metadata, db_constructor, db_params,
42
  config
43
  )
44
  return predictions
@@ -62,7 +63,7 @@ def predict():
62
  model_generative,
63
  model_classifier, classifier_head,
64
  tokenizer_generative, tokenizer_classifier,
65
- db_metadata, db_constructor, db_params,
66
  config
67
  )
68
 
 
13
  db_metadata = components[0]
14
  db_constructor = components[1]
15
  db_params = components[2]
16
+ ex_list = components[3]
17
+ model_retrieval = components[4]
18
+ model_generative = components[5]
19
+ tokenizer_generative = components[6]
20
+ model_classifier = components[7]
21
+ classifier_head = components[8]
22
+ tokenizer_classifier = components[9]
23
 
24
  def call_predict_api(
25
  input_query,
 
27
  model_generative,
28
  model_classifier, classifier_head,
29
  tokenizer_generative, tokenizer_classifier,
30
+ db_metadata, db_constructor, db_params, ex_list,
31
  config
32
  ):
33
  '''
 
39
  model_generative,
40
  model_classifier, classifier_head,
41
  tokenizer_generative, tokenizer_classifier,
42
+ db_metadata, db_constructor, db_params, ex_list,
43
  config
44
  )
45
  return predictions
 
63
  model_generative,
64
  model_classifier, classifier_head,
65
  tokenizer_generative, tokenizer_classifier,
66
+ db_metadata, db_constructor, db_params, ex_list,
67
  config
68
  )
69