Spaces:
Runtime error
Runtime error
dalexanderch
commited on
Commit
•
6fa8fad
1
Parent(s):
bdac097
Upload app.py
Browse files
app.py
CHANGED
@@ -53,7 +53,7 @@ def fn(model, class_list):
|
|
53 |
def f(glycan):
|
54 |
glycan = [glycan]
|
55 |
label = [0]
|
56 |
-
data = next(iter(dataset_to_dataloader(glycan, label, batch_size=1)))
|
57 |
device = "cpu"
|
58 |
if torch.cuda.is_available():
|
59 |
device = "cuda:0"
|
|
|
53 |
def f(glycan):
|
54 |
glycan = [glycan]
|
55 |
label = [0]
|
56 |
+
data = next(iter(dataset_to_dataloader(glycan, label, batch_size=1, libr=lib)))
|
57 |
device = "cpu"
|
58 |
if torch.cuda.is_available():
|
59 |
device = "cuda:0"
|