dalexanderch commited on
Commit
853a85f
1 Parent(s): 00a6701

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -69,7 +69,7 @@ def fn(glycan, model):
69
  label = [0]
70
  data = next(iter(dataset_to_dataloader(glycan, label, batch_size=1)))
71
 
72
- if model == "Ensemble":
73
  pred = model_pred(data)
74
  else:
75
  device = "cpu"
 
69
  label = [0]
70
  data = next(iter(dataset_to_dataloader(glycan, label, batch_size=1)))
71
 
72
+ if model in ["Ensemble", "Bootstrap Ensemble"]:
73
  pred = model_pred(data)
74
  else:
75
  device = "cpu"