Update app.py

app.py

@@ -72,6 +72,7 @@ def classification(text, k):
     text = gbif_normalization(text)
     result = classification_model(text)
     habitat_labels = [res['label'] for res in result[0][:k]]
+    habitat_name = eunis_habitats[eunis_habitats['EUNIS 2020 code'] == habitat_labels[0]]['EUNIS-2021 habitat name'].values[0]
     if k == 1:
         text = f"This vegetation plot probably belongs to the habitat type {habitat_labels[0]}."
         text += f"\nThis habitat type is named '{habitat_name}'."
@@ -81,7 +82,6 @@ def classification(text, k):
     else:
         text = f"This vegetation plot probably belongs to the habitat type {', '.join(habitat_labels[:-1])}, or {habitat_labels[-1]}."
         text += f"\nThe most likely habitat type (i.e., habitat_labels[0]) is named '{habitat_name}'."
-    habitat_name = eunis_habitats[eunis_habitats['EUNIS 2020 code'] == habitat_labels[0]]['EUNIS-2021 habitat name'].values[0]
     text += f"\nSee an image of this habitat type below."
     image_output = return_habitat_image(habitat_labels[0])
     return text, image_output