Update app.py

app.py

@@ -119,7 +119,8 @@ if disease_or_gard_id:
                                 NER_pipeline, entity_classes,
                                 extract_diseases, GARD_dict, max_length,
                                 classify_model_vars)
-    if df:
+    #IF it returns something, then continue.
+    if sankey_data:
         df.replace(to_replace='None', value="None")
         st.dataframe(df, height=200)
         csv = convert_df(df)