Update app.py

app.py

@@ -77,6 +77,7 @@ 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))
@@ -85,7 +86,14 @@ def detect_lang(src_sentence):
     Y_logits = model(torch.Tensor(X_test))
     pred_languages = torch.argmax(Y_logits, dim=-1).tolist()
     return list(map(idx2lang.get, pred_languages))[0]
-
+'''
+# sklearn
+def detect_lang(src_sentence):
+    X_test = vectorizer.transform([build_freq_dict(src_sentence)])
+    # predict using sklearn
+    pred_idx = clf.predict(X_test)[0]
+    return idx2lang[pred_idx]
+    
 # --- Gradio UI ---
 with gr.Blocks(title="language detector") as demo:
     gr.Markdown("# language detector")