Update app.py

app.py

@@ -69,7 +69,7 @@ def AMP(file):
             # out_probability.extend(np.array(predict.cpu())[:, -1].tolist())
             test_argmax = np.argmax(predict.cpu(), axis=1).tolist()
     id2str = {0:"non-AMP", 1:"AMP"}
-    return id2str[test_argmax], out_probability
+    return id2str[test_argmax[0]], out_probability
 
 iface = gr.Interface(fn=AMP,
                      inputs="text",