Update app.py

app.py

@@ -81,7 +81,7 @@ def extract_features(sequence):
 
 def predict(sequence):
     """Predict if the sequence is an AMP or not."""
-    features = extract_features(sequence)
+    features = np.array(features).reshape(1, -1)  # Reshape for a single sample
     prediction = model.predict(features)[0]
     probabilities = model.predict_proba(features)[0]