Update app.py

app.py

@@ -61,7 +61,6 @@ def build_freq_dict(sentence, MAXES=MAXES, MAX_SHIFT=MAX_SHIFT):
     return shift_keys(fhcodes, MAX_SHIFT)
 
 # --- load models ---
-clf = joblib.load("nld.joblib")
 vectorizer = joblib.load("nld_vectorizer.joblib")
 idx2lang = joblib.load("nld_lang_codes.joblib")
 
@@ -77,26 +76,33 @@ model.load_state_dict(torch.load("nld.pth", map_location="cpu"))
 model.eval()
 
 # --- prediction function ---
-def detect_lang(src_sentence):
-    src_sentence = [src_sentence]
-    X_test = vectorizer.transform(map(build_freq_dict, src_sentence))
+def detect_lang(src_sentence: str):
+    feat_dict = build_freq_dict(src_sentence)
+    X_test = vectorizer.transform([feat_dict])  # ✅ 用 list 包起来
     if hasattr(X_test, "toarray"):
         X_test = X_test.toarray()
-    Y_logits = model(torch.Tensor(X_test))
-    pred_languages = torch.argmax(Y_logits, dim=-1).tolist()
-    return list(map(idx2lang.get, pred_languages))[0]
+    X_test = torch.from_numpy(X_test.astype("float32"))
+
+    with torch.no_grad():
+        logits = model(X_test)
+        pred_idx = torch.argmax(logits, dim=-1).item()
+    return idx2lang[pred_idx]
 
 # --- Gradio UI ---
-with gr.Blocks(title="ZheZhanglanguage detector") as demo:
-    gr.Markdown("# ZheZhang language detector")
+with gr.Blocks(title="Antons language detector") as demo:
+    gr.Markdown("# Antons language detector")
     with gr.Row():
         with gr.Column():
             src_sentence = gr.Textbox(
-                label="Text", placeholder="Write your text...")
+                label="Text", placeholder="Write your text..."
+            )
         with gr.Column():
             tgt_sentence = gr.Textbox(
-                label="Language", placeholder="Language will show here...")
+                label="Language",
+                placeholder="Language will show here...",
+                interactive=False  # ✅ 输出框不可编辑
+            )
     btn = gr.Button("Guess the language!")
     btn.click(fn=detect_lang, inputs=[src_sentence], outputs=[tgt_sentence])
 
-demo.launch()
+demo.launch()