Update app.py

app.py

@@ -60,14 +60,7 @@ def predict(sequence):
     prediction = model.predict(features.T)[0]
     probability_amp = model.predict_proba(features.T)
     
-    if prediction == 0:
-        prediction_label = "Potential Bioactive Peptide with Antimicrobial Properties (P-AMP)"
-        probability_amp = probabilities[0] 
-    else:
-        prediction_label = "Likely Non-Antimicrobial Peptide"
-        probability_amp = probabilities[0]
-
-    return prediction_label, probability_amp 
+    return "Potential Bioactive Peptide with Antimicrobial Properties (P-AMP)" if prediction == 0 else "Likely Non-Antimicrobial Peptide"
 
 
 iface = gr.Interface(