Update app.py

app.py

@@ -72,12 +72,65 @@ def predict_mushroom(*inputs):
     # Return the human-readable result
     return 'Edible' if prediction[0] == 0 else 'Poisonous'
 
+# Example 1: Death cap mushroom (poisonous)
+death_cap_example = [
+    "convex",            # cap-shape ('x')
+    "smooth",            # cap-surface ('s')
+    "green",             # cap-color ('r') – assuming 'green' is in our options for cap-color
+    "bruises",           # bruises ('t')
+    "none",              # odor ('n')
+    "free",              # gill-attachment ('f')
+    "distant",           # gill-spacing ('d')
+    "narrow",            # gill-size ('n')
+    "white",             # gill-color ('w')
+    "tapering",          # stalk-shape ('t')
+    "cup",               # stalk-root ('u')
+    "smooth",            # stalk-surface-above-ring ('s')
+    "smooth",            # stalk-surface-below-ring ('s')
+    "white",             # stalk-color-above-ring ('w')
+    "white",             # stalk-color-below-ring ('w')
+    "universal",         # veil-type ('u')
+    "white",             # veil-color ('w')
+    "one",               # ring-number ('o')
+    "evanescent",        # ring-type ('e')
+    "white",             # spore-print-color ('w')
+    "scattered",         # population ('s')
+    "woods"              # habitat ('d')
+]
+
+# Example 2: Normal brown edible mushroom
+edible_example = [
+    "bell",              # cap-shape ('b')
+    "fibrous",           # cap-surface ('f')
+    "brown",             # cap-color ('n')
+    "no",                # bruises ('f')
+    "almond",            # odor ('a')
+    "attached",          # gill-attachment ('a')
+    "close",             # gill-spacing ('c')
+    "broad",             # gill-size ('b')
+    "brown",             # gill-color ('n')
+    "enlarging",         # stalk-shape ('e')
+    "bulbous",           # stalk-root ('b')
+    "fibrous",           # stalk-surface-above-ring ('f')
+    "fibrous",           # stalk-surface-below-ring ('f')
+    "brown",             # stalk-color-above-ring ('n')
+    "brown",             # stalk-color-below-ring ('n')
+    "partial",           # veil-type ('p')
+    "brown",             # veil-color ('n')
+    "none",              # ring-number ('n')
+    "none",              # ring-type ('n')
+    "brown",             # spore-print-color ('n')
+    "numerous",          # population ('n')
+    "meadows"            # habitat ('m')
+]
+
 demo = gr.Interface(
     fn=predict_mushroom,
     inputs=[gr.Dropdown(choices=list(options.values()), label=feature) for feature, options in feature_options.items()],
-    outputs="text",
+    outputs=gr.Textbox(label="Prediction"),
     title="MycoNom - Mushroom Edibility Classifier",
-    description="Select the mushroom features to determine if it's edible or poisonous.<br><br>You can train your own version of this model by heading to OPEN-ARC: https://github.com/Infinitode/OPEN-ARC.<br><br>**Disclaimer:** This model is for **educational purposes only** and should not be used for real-life mushroom classification or any decision-making processes related to the consumption of mushrooms. While the model performs well on the provided dataset, it has not been thoroughly validated for real-world scenarios and may not accurately detect poisonous mushrooms in all conditions. Always consult an expert or use trusted resources when identifying mushrooms."
+    description="Select the mushroom features to determine if it's edible or poisonous.<br><br>You can train your own version of this model by heading to OPEN-ARC: https://github.com/Infinitode/OPEN-ARC.<br><br>**Disclaimer:** This model is for **educational purposes only** and should not be used for real-life mushroom classification or any decision-making processes related to the consumption of mushrooms. While the model performs well on the provided dataset, it has not been thoroughly validated for real-world scenarios and may not accurately detect poisonous mushrooms in all conditions. Always consult an expert or use trusted resources when identifying mushrooms.",
+    examples=[death_cap_example, edible_example]
 )
 
 demo.launch()