Upload app.py

app.py

@@ -14,6 +14,9 @@ def fn(class_list):
     if model == "No data augmentation":
       model = torch.load("model1.pt", map_location=torch.device('cpu'))
       model.eval()
+    elif model == "Ensemble":
+      model = torch.load("model3.pt", map_location=torch.device('cpu'))
+      model.eval()
     else:
       model = torch.load("model2.pt", map_location=torch.device('cpu'))
       model.eval()
@@ -48,6 +51,7 @@ demo = gr.Interface(
     allow_flagging=False,
     title="SweetNet demo",
     examples=[["GlcOSN(a1-4)GlcA(b1-4)GlcOSN(a1-4)GlcAOS(b1-4)GlcOSN(a1-4)GlcOSN", "No data augmentation"],
-    ["Man(a1-2)Man(a1-3)[Man(a1-3)Man(a1-6)]Man(b1-4)GlcNAc(b1-4)GlcNAc", "Random node deletion"]]
+    ["Man(a1-2)Man(a1-3)[Man(a1-3)Man(a1-6)]Man(b1-4)GlcNAc(b1-4)GlcNAc", "Random node deletion"]
+    ["Man(a1-2)Man(a1-3)[Man(a1-6)]Man(a1-6)[Man(a1-2)Man(a1-2)Man(a1-3)]Man(b1-4)GlcNAc", "Ensemble"]]
 )
 demo.launch(debug=True)