Spaces:
Runtime error
Runtime error
Commit
•
853a85f
1
Parent(s):
00a6701
Update app.py
Browse files
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
|
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"
|