Spaces:
Runtime error
Runtime error
Commit
·
7ec6647
1
Parent(s):
b8c4fc1
Update app.py
Browse files
app.py
CHANGED
@@ -81,7 +81,7 @@ def fn(glycan, model):
|
|
81 |
label = [0]
|
82 |
data = next(iter(dataset_to_dataloader(glycan, label, batch_size=1)))
|
83 |
|
84 |
-
if model in ["Ensemble", "Bootstrap
|
85 |
pred = model_pred(data)
|
86 |
else:
|
87 |
device = "cpu"
|
|
|
81 |
label = [0]
|
82 |
data = next(iter(dataset_to_dataloader(glycan, label, batch_size=1)))
|
83 |
|
84 |
+
if model in ["Ensemble", "Bootstrap ensemble"]:
|
85 |
pred = model_pred(data)
|
86 |
else:
|
87 |
device = "cpu"
|