Update app.py

app.py

@@ -65,6 +65,9 @@ def fn(glycan, model):
     elif model == "Random edge deletion":
       model_pred = model5
       model_pred.eval()
+    elif model == "Hierarchy substitution"
+      model_pred = model6
+      model_pred.eval()
     else:
       model_pred = model2
       model_pred.eval()
@@ -93,7 +96,7 @@ def fn(glycan, model):
 
 demo = gr.Interface(
     fn=fn,
-    inputs=[gr.Textbox(label="Glycan sequence", value="Man(a1-2)Man(a1-3)[Man(a1-3)Man(a1-6)]Man(b1-4)GlcNAc(b1-4)GlcNAc"), gr.Radio(label="Model",choices=["No data augmentation", "Random node deletion", "Random edge deletion", "Ensemble", "Bootstrap Ensemble"])],
+    inputs=[gr.Textbox(label="Glycan sequence", value="Man(a1-2)Man(a1-3)[Man(a1-3)Man(a1-6)]Man(b1-4)GlcNAc(b1-4)GlcNAc"), gr.Radio(label="Model",choices=["No data augmentation", "Random node deletion", "Random edge deletion", "Ensemble", "Bootstrap Ensemble", "Hierarchy substitution"])],
     outputs=[gr.Label(num_top_classes=15, label="Prediction"), gr.Image(label="Glycan graph")],
     allow_flagging="never",
     title="SweetNet demo",