Update app.py

app.py

@@ -47,7 +47,7 @@ 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).reshape(-1,1)
+    feature_array = np.array(selected_feature_values).reshape(1,-1)
     # Min-Max Normalization
     scaler = MinMaxScaler()
     normalized_features = scaler.fit_transform(feature_array).flatten()
@@ -59,7 +59,7 @@ def predict(sequence):
     """Predict AMP vs Non-AMP"""
     features = extract_features(sequence)
     prediction = model.predict(features)[0]
-    return "AMP" if prediction == 1 else "Non-AMP"
+    return "AMP" if prediction == 0 else "Non-AMP"
 
 # Create Gradio interface
 iface = gr.Interface(