Update app.py

app.py

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