Update app.py

app.py

@@ -10,15 +10,21 @@ scaler = MinMaxScaler()
 
 def extract_features(sequence):
     """Calculate AAC, Dipeptide Composition, and normalize features."""
-    # Calculate Amino Acid Composition (AAC)
-    aac = AAComposition.CalculateAADipeptideComposition(sequence)
+    # Calculate Amino Acid Composition (AAC) and convert to array
+    aac = np.array(list(propy.AAComposition.CalculateAAC(sequence).values()))
+
+    # Calculate Dipeptide Composition and convert to array
+    dipeptide_comp = np.array(list(propy.AAComposition.CalculateAADipeptideComposition(sequence).values()))
+
+    # Combine both features (AAC and Dipeptide Composition)
+    features = np.concatenate((aac, dipeptide_comp))
+
+    # Normalize using the pre-trained scaler (Ensure the scaler is loaded correctly)
+    normalized_features = scaler.transform([features])  # Don't use fit_transform(), only transform()
 
-    
-    # Normalize with pre-trained scaler (avoid fitting new data)
-    normalized_features = scaler.fit_transform([aac])
-    
     return normalized_features
 
+
 def predict(sequence):
     """Predict AMP vs Non-AMP"""
     features = extract_features(sequence)