Update app.py

app.py

@@ -1,31 +1,5 @@
-from huggingface_hub import hf_hub_download
-import fasttext
-import numpy as np
-import gradio as gr
+# Load model directly
+from transformers import AutoTokenizer, AutoModelForQuestionAnswering
 
-# 모델 다운로드
-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 = patched_predict(model, text)
-    return {
-        "Predicted language": predictions[0][0],
-        "Confidence score": predictions[1][0]
-    }
-
-# Gradio 인터페이스
-interface = gr.Interface(
-    fn=predict_language,
-    inputs=gr.Textbox(label="Input Text"),
-    outputs="json",
-    title="Language Predictor"
-)
-
-interface.launch()
+tokenizer = AutoTokenizer.from_pretrained("ainize/klue-bert-base-mrc")
+model = AutoModelForQuestionAnswering.from_pretrained("ainize/klue-bert-base-mrc")