Update app.py

app.py

@@ -68,7 +68,9 @@ def predict(sequence):
     """Predict AMP vs Non-AMP"""
     features = extract_features(sequence)
     prediction = model.predict(features.T)[0]
-    return "Potential Bioactive Peptide with Antimicrobial Properties (P-AMP)" if prediction == 0 else "Likely Non-Antimicrobial Peptide"
+    probabilities = model.predict_proba(features.T)
+
+    return "Potential Bioactive Peptide with Antimicrobial Properties (P-AMP)" if prediction == 0 else "Likely Non-Antimicrobial Peptide", probabilities[0]
 
 # Create Gradio interface
 iface = gr.Interface(