Update app.py

app.py

@@ -47,8 +47,7 @@ def AMP(file):
     model.eval()
     out_probability = []
     with torch.no_grad():
-        batchs = {k: v for k, v in batch.items()}
-        predict = model(batchs)
+        predict = model(test_data)
         out_probability.extend(np.max(np.array(predict.cpu()),axis=1).tolist())
         test_argmax = np.argmax(predict.cpu(), axis=1).tolist()
     id2str = {0:"non-AMP", 1:"AMP"}