Adding threshold

app.py

@@ -70,13 +70,13 @@ def detect_lang(
         return predictions if return_proba else predictions[0][0]
 
 
-def identify(text):
-    return {language_dict[lang]: proba for lang, proba in detect_lang(text.replace("\n", " "), return_proba=True)}
+def identify(text, threshold):
+    return {language_dict[lang]: proba for lang, proba in detect_lang(text.replace("\n", " "), threshold=threshold / 100.0, return_proba=True)}
 
 iface = gr.Interface(
     title="NB Nordic Language Identification",
     description="""This demo uses the [NB-Nordic-LID](https://huggingface.co/NbAiLab/nb-nordic-lid) model to classify a given text into one of the 12 Nordic languages supported. <b>At least 3 or 4 words are needed to identify the language.</b>""",
     fn=identify,
-    inputs="text", 
+    inputs=[gr.Textbox(label="Text to identify language for"), gr.Slider(0, 100, value=50, step=1, label="Probability threshold (%)")], 
     outputs="label")
 iface.launch()