dalexanderch commited on
Commit
6fa8fad
1 Parent(s): bdac097

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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"