Update app.py

app.py

@@ -12,12 +12,13 @@ app = Flask(__name__)
 components = initialize_all_components(config)
 db_metadata = components[0]
 db_constructor = components[1]
-model_retrieval = components[2]
-model_generative = components[3]
-tokenizer_generative = components[4]
-model_classifier = components[5]
-classifier_head = components[6]
-tokenizer_classifier = components[7]
+db_params = components[2]
+model_retrieval = components[3]
+model_generative = components[4]
+tokenizer_generative = components[5]
+model_classifier = components[6]
+classifier_head = components[7]
+tokenizer_classifier = components[8]
 
 def call_predict_api(
     input_query, 
@@ -25,7 +26,7 @@ def call_predict_api(
     model_generative,  
     model_classifier, classifier_head,
     tokenizer_generative, tokenizer_classifier,
-    db_metadata, db_constructor,
+    db_metadata, db_constructor, db_params,
     config
     ):
     '''
@@ -37,7 +38,7 @@ def call_predict_api(
         model_generative,  
         model_classifier, classifier_head,
         tokenizer_generative, tokenizer_classifier,
-        db_metadata, db_constructor,
+        db_metadata, db_constructor, db_params,
         config
     )
     return predictions
@@ -61,7 +62,7 @@ def predict():
                 model_generative,
                 model_classifier, classifier_head,
                 tokenizer_generative, tokenizer_classifier,
-                db_metadata, db_constructor,
+                db_metadata, db_constructor, db_params,
                 config
             )
         print("success prediction!")