app.py

@@ -62,7 +62,7 @@ def gbif_normalization(text):
     text = text.lower()
     return text
 
-def classification(text):
+def classification(text, k):
     text = gbif_normalization(text)
     result = classification_model(text)
     habitat_label = result[0]['label']