Upload app.py

app.py

@@ -45,7 +45,7 @@ def fn(glycan, model):
     glycan = [glycan]
     label = [0]
     data = next(iter(dataset_to_dataloader(glycan, label, batch_size=1)))
-    pred = model(data)[0]
+    pred = model(data)
     pred = np.exp(pred)/sum(np.exp(pred)) # Softmax 
     pred = [float(x) for x in pred]
     pred = {class_list[i]:pred[i] for i in range(15)}