Update app.py

app.py

@@ -64,20 +64,22 @@ def load_models():
 
 def epi_sankey(sankey_data):
     gathered, relevant, epidemiologic = sankey_data
+    
     fig = go.Figure(data=[go.Sankey(
-    node = dict(
-      pad = 15,
-      thickness = 20,
-      line = dict(color = "black", width = 0.5),
-      label = ["PubMed IDs Gathered", "Relevant Abstracts Gathered", "Irrelevant", "Epidemiologic", "Irrelevant"],
-      color = "blue"
-    ),
-    #label = ["A1", "A2", "B1", "B2", "C1", "C2"]
-    link = dict(
-      source = [0, 0, 0, 1, 3],
-      target = [0, 1, 2, 3, 4],
-      value = [gathered, gathered-relevant, relevant, epidemiologic, gathered-epidemiologic]
-  ))])
+        node = dict(
+          pad = 15,
+          thickness = 20,
+          line = dict(color = "black", width = 0.5),
+          label = ["PubMed IDs Gathered", "Relevant Abstracts Gathered", "Irrelevant", "Epidemiologic", "Irrelevant"],
+          color = "blue"
+        ),
+        #label = ["A1", "A2", "B1", "B2", "C1", "C2"]
+        link = dict(
+          source = [0, 0, 0, 1, 3],
+          target = [0, 1, 2, 3, 4],
+          value = [gathered, gathered-relevant, relevant, epidemiologic, gathered-epidemiologic]
+      ))])
+    
     return fig
 
 with st.spinner('Loading Epidemiology Models and Dependencies...'):