Update app.py

app.py

@@ -3,6 +3,7 @@ import fasttext
 from huggingface_hub import hf_hub_download
 import re
 import string
+import numpy as np
 
 def load_GlotLID():
     model_path = hf_hub_download(repo_id="cis-lmu/glotlid", filename="model_v3.bin")
@@ -23,10 +24,11 @@ def compute(sentence):
     sentence = preprocess_text(sentence)
     
     # Get top 3 predictions
-    output = model.predict(sentence, k=3)
+    labels, probs = model.predict(sentence, k=3)
+    probs = np.asarray(probs)
     
     results = []
-    for label, score in zip(output[0], output[1]):
+    for label, score in zip(labels, probs):
         label = label.split('__')[-1]
         results.append(f"{label}: {score:.4f}")