Update app.py

app.py

@@ -78,10 +78,7 @@ def extract_features(sequence):
     normalized_features = normalized_features.flatten() # Flatten AFTER normalization if needed
 
 
-    selected_feature_dict = {}
-    for i, feature in enumerate(selected_features):
-        if feature in all_features_dict:
-            selected_feature_dict[feature] = normalized_features[i]
+    selected_feature_dict = {feature: normalized_features[i] for i, feature in enumerate(selected_features) if feature in all_features}
 
     selected_feature_df = pd.DataFrame([selected_feature_dict])
     selected_feature_array = selected_feature_df.T.to_numpy()