Update app.py

app.py

@@ -36,6 +36,7 @@ class_list=['Amoebozoa', 'Animalia', 'Bacteria', 'Bamfordvirae', 'Chromista', 'E
 model1 = torch.load("model1.pt", map_location=torch.device('cpu'))
 model2 = torch.load("model2.pt", map_location=torch.device('cpu'))
 model3 = torch.load("model3.pt", map_location=torch.device('cpu'))
+model4 = torch.load("model4.pt", map_location=torch.device('cpu'))
 
 def fn(glycan, model):
     # Draw graph
@@ -57,6 +58,9 @@ def fn(glycan, model):
     elif model == "Ensemble":
       model_pred = model3
       model_pred.eval()
+    elif model == "Bootstrap Ensemble":
+      model_pred = model4
+      model_pred.eval()
     else:
       model_pred = model2
       model_pred.eval()
@@ -85,7 +89,7 @@ def fn(glycan, model):
 
 demo = gr.Interface(
     fn=fn,
-    inputs=[gr.Textbox(label="Glycan sequence"), gr.Radio(label="Model",choices=["No data augmentation", "Random node deletion", "Ensemble"])],
+    inputs=[gr.Textbox(label="Glycan sequence"), gr.Radio(label="Model",choices=["No data augmentation", "Random node deletion", "Ensemble", "Bootstrap Ensemble"])],
     outputs=[gr.Label(num_top_classes=15, label="Prediction"), gr.Image(label="Glycan graph")],
     allow_flagging="never",
     title="SweetNet demo",