Update app.py

app.py

@@ -47,10 +47,10 @@ def extract_features(sequence):
     selected_feature_values = [all_features[feature] for feature in selected_features if feature in all_features]
 
     # Convert to NumPy array for normalization
-    feature_array = np.array(selected_feature_values)
+    feature_array = np.array(selected_feature_values).reshape(-1, 1)
     # Min-Max Normalization
     scaler = MinMaxScaler()
-    normalized_features = scaler.fit_transform(feature_array).flatten()
+    normalized_features = scaler.fit_transform(feature_array)
 
     return normalized_features
 
@@ -58,7 +58,7 @@ def extract_features(sequence):
 def predict(sequence):
     """Predict AMP vs Non-AMP"""
     features = extract_features(sequence)
-    prediction = model.predict(features)[0]
+    prediction = model.predict(features.T)[0]
     return "AMP" if prediction == 0 else "Non-AMP"
 
 # Create Gradio interface
@@ -71,4 +71,4 @@ iface = gr.Interface(
 )
 
 # Launch app
-iface.launch(share=True)
+iface.launch(share=True)