Update app.py

app.py

@@ -1,16 +1,20 @@
 from huggingface_hub import hf_hub_download
 import fasttext
+import numpy as np
 import gradio as gr
 
 # 모델 다운로드
 model_path = hf_hub_download(repo_id="cis-lmu/glotlid", filename="model.bin")
-
-# 모델 로드
 model = fasttext.load_model(model_path)
 
+# 수정된 predict 함수
+def patched_predict(model, text):
+    labels, probs = model.predict(text)
+    return labels, np.asarray(probs)
+
 # 예측 함수
 def predict_language(text):
-    predictions = model.predict(text)
+    predictions = patched_predict(model, text)
     return {
         "Predicted language": predictions[0][0],
         "Confidence score": predictions[1][0]