Update app.py

app.py

@@ -31,8 +31,17 @@ feature_options = {
     'habitat': {'grasses': 'g', 'leaves': 'l', 'meadows': 'm', 'paths': 'p', 'urban': 'u', 'waste': 'w', 'woods': 'd'}
 }
 
-def predict_mushroom(features):
-    numerical_features = {feature: feature_options[feature][value] for feature, value in features.items()}
+# def predict_mushroom(features):
+#     numerical_features = {feature: feature_options[feature][value] for feature, value in features.items()}
+#     input_df = pd.DataFrame([numerical_features])
+#     prediction = model.predict(input_df)
+#     return 'Poisonous' if prediction[0] == 1 else 'Edible'
+
+def predict_mushroom(*features):
+    # Convert the feature names to a dictionary based on the order
+    feature_names = list(feature_options.keys())
+    numerical_features = {feature_names[i]: feature_options[feature_names[i]][value] for i, value in enumerate(features)}
+    
     input_df = pd.DataFrame([numerical_features])
     prediction = model.predict(input_df)
     return 'Poisonous' if prediction[0] == 1 else 'Edible'