Dynamic loading (#8)

- Dynamic loading (91fb880bb0580b4b29a2bf626bbc304a6868e7b5)


Co-authored-by: H Wilks <[email protected]>

app.py

@@ -36,42 +36,42 @@ PIPELINES = [
     {
         'id': 1,
         'name': 'Baseline',
-        'pipeline': load_pipeline_from_hub("pipeline_ex1_s1.joblib")
+        'filename': "pipeline_ex1_s1.joblib"
     },
     {
         'id': 2,
         'name': 'Trained on a FeedForward NN',
-        'pipeline': load_pipeline_from_hub("pipeline_ex1_s2.joblib")
+        'filename': "pipeline_ex1_s2.joblib"
     },
     {
         'id': 3,
         'name': 'Trained on a CRF',
-        'pipeline': load_pipeline_from_hub("pipeline_ex1_s2.joblib")
+        'filename': "pipeline_ex1_s2.joblib"
     },
     {
         'id': 4,
         'name': 'Trained on a small dataset',
-        'pipeline': load_pipeline_from_hub("pipeline_ex2_s3.joblib")
-    },
+        'filename': "pipeline_ex2_s3.joblib"
+v    },
     {
         'id': 5,
         'name': 'Trained on a large dataset',
-        'pipeline': load_pipeline_from_hub("pipeline_ex2_s2.joblib")
+        'filename': "pipeline_ex2_s2.joblib"
     },
     {
         'id': 6,
         'name': 'Embedded using TFIDF',
-        'pipeline': load_pipeline_from_hub("pipeline_ex3_s2.joblib")
+        'filename': "pipeline_ex3_s2.joblib"
     },
     {
         'id': 7,
         'name': 'Embedded using GloVe',
-        'pipeline': load_pipeline_from_hub("pipeline_ex3_s3.joblib")
+        'filename': "pipeline_ex3_s3.joblib"
     },
     {
          'id': 8,
          'name': 'Embedded using Bio2Vec',
-        'pipeline': load_pipeline_from_hub("pipeline_ex3_s4.joblib")
+        'filename': "pipeline_ex3_s4.joblib"
     },
     
 ]
@@ -79,7 +79,7 @@ PIPELINES = [
 pipeline_metadata = [{'id': p['id'], 'name': p['name']} for p in PIPELINES]
 
 def get_pipeline_by_id(pipelines, pipeline_id):
-    return next((p['pipeline'] for p in pipelines if p['id'] == pipeline_id), None)
+    return next((p['filename'] for p in pipelines if p['id'] == pipeline_id), None)
 
 def get_name_by_id(pipelines, pipeline_id):
     return next((p['name'] for p in pipelines if p['id'] == pipeline_id), None)
@@ -130,7 +130,7 @@ def get_data():
         tokens_fomatted = pd.Series([pd.Series(tokens)])
 
         pipeline_id = int(request.form['pipeline_select'])
-        pipeline = get_pipeline_by_id(PIPELINES, pipeline_id)
+        pipeline = load_pipeline_from_hub(get_pipeline_by_id(PIPELINES, pipeline_id))
         name = get_name_by_id(PIPELINES, pipeline_id)
         
         labels = requestResults(tokens_fomatted, pipeline)