Update app.py

app.py

@@ -43,12 +43,14 @@ def predict_mushroom(*inputs):
     features = list(feature_options.keys())
     user_input = dict(zip(features, inputs))
 
-    # Convert string inputs to numerical values using mappings
+    # Convert the string inputs (full names) directly to corresponding letters using the mappings
     numerical_features = {}
     for feature, value in user_input.items():
         if feature in mappings:
-            if value in mappings[feature]:
-                numerical_features[feature] = mappings[feature][value]  # Map string to numerical value
+            # Reverse the mapping to find the corresponding letter
+            inverse_mapping = {v: k for k, v in mappings[feature].items()}
+            if value in inverse_mapping:
+                numerical_features[feature] = inverse_mapping[value]  # Map full name to letter
             else:
                 raise ValueError(f"Invalid value '{value}' for feature '{feature}'.")
         else:
@@ -71,7 +73,7 @@ demo = gr.Interface(
     inputs=[gr.Dropdown(choices=list(options.values()), label=feature) for feature, options in feature_options.items()],
     outputs="text",
     title="MycoNom - Mushroom Edibility Classifier",
-    description="Select the mushroom features to determine if it's edible or poisonous.<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."
 )
 
 demo.launch()