Update app.py

app.py

@@ -1,6 +1,7 @@
 import fasttext
 from huggingface_hub import hf_hub_download
 import gradio as gr
+import numpy as np
 
 model_path = hf_hub_download(repo_id="facebook/fasttext-language-identification", filename="model.bin")
 model = fasttext.load_model(model_path)
@@ -8,7 +9,7 @@ model = fasttext.load_model(model_path)
 def predict(text, top):
     labels, probabilities = model.predict(text, k=top)
     cleaned_labels = [label.replace('__label__', '') for label in labels]
-    result = dict(zip(cleaned_labels, probabilities))
+    result = dict(zip(cleaned_labels, np.array(probabilities)))
     #result = sorted(result, key=lambda x: x[1], reverse=True)
     return result