dalexanderch commited on
Commit
af8a4b3
1 Parent(s): e9847e4

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -58,10 +58,10 @@ def fn(glycan, model):
58
  if model == "No data augmentation":
59
  model_pred = model1
60
  model_pred.eval()
61
- elif model == "Ensemble":
62
  model_pred = model3
63
  model_pred.eval()
64
- elif model == "Bootstrap ensemble":
65
  model_pred = model4
66
  model_pred.eval()
67
  elif model == "Random edge deletion":
@@ -108,8 +108,8 @@ demo = gr.Interface(
108
  examples=[
109
  ["D-Rha(b1-2)D-Rha(b1-2)Gal(b1-4)[Glc(b1-2)]GlcAOMe", "Random node deletion"],
110
  ["Neu5Ac(a2-3)Gal(b1-4)GlcNAc(b1-3)GalNAc", "No data augmentation"],
111
- ["Kdo(a2-4)[Kdo(a2-8)]Kdo(a2-4)Kdo", "Ensemble"],
112
- ["Galf(b1-6)Galf(b1-5)Galf(b1-6)Galf", "Bootstrap Ensemble"],
113
  ["GlcNAc(b1-2)Rha(a1-2)Rha(b1-3)Rha(a1-3)GlcNAc", "Random edge deletion"],
114
  ["Pse(b2-6)Glc(b1-6)Gal(b1-3)GalNAc(b1-3)[Glc(b1-6)]Gal(b1-3)GalNAc", "Adjusted class weights"],
115
  ]
 
58
  if model == "No data augmentation":
59
  model_pred = model1
60
  model_pred.eval()
61
+ elif model == "Classical Ensemble":
62
  model_pred = model3
63
  model_pred.eval()
64
+ elif model == "Bagging ensemble":
65
  model_pred = model4
66
  model_pred.eval()
67
  elif model == "Random edge deletion":
 
108
  examples=[
109
  ["D-Rha(b1-2)D-Rha(b1-2)Gal(b1-4)[Glc(b1-2)]GlcAOMe", "Random node deletion"],
110
  ["Neu5Ac(a2-3)Gal(b1-4)GlcNAc(b1-3)GalNAc", "No data augmentation"],
111
+ ["Kdo(a2-4)[Kdo(a2-8)]Kdo(a2-4)Kdo", "Classical ensemble"],
112
+ ["Galf(b1-6)Galf(b1-5)Galf(b1-6)Galf", "Bagging Ensemble"],
113
  ["GlcNAc(b1-2)Rha(a1-2)Rha(b1-3)Rha(a1-3)GlcNAc", "Random edge deletion"],
114
  ["Pse(b2-6)Glc(b1-6)Gal(b1-3)GalNAc(b1-3)[Glc(b1-6)]Gal(b1-3)GalNAc", "Adjusted class weights"],
115
  ]