Update app.py

app.py

@@ -33,9 +33,18 @@ def load_models():
         use_safetensors=True
     ).to(device)
 
-    from keybert.backend._hf import HFEmbeddingBackend
-    kb_backend = HFEmbeddingBackend(pubmed_model, pubmed_tok)
-    kb_model   = KeyBERT(model=kb_backend)
+    # from keybert.backend._hf import HFEmbeddingBackend
+    # kb_backend = HFEmbeddingBackend(pubmed_model, pubmed_tok)
+    # kb_model   = KeyBERT(model=kb_backend)
+    try:
+        # Способ 1: Используем TransformerBackend (новый способ)
+        from keybert.backend import TransformerBackend
+        custom_embedder = TransformerBackend(pubmed_model, pubmed_tok)
+        kb_model = KeyBERT(model=custom_embedder)
+    except ImportError:
+        # Способ 2: Fallback - используем стандартную модель
+        kb_model = KeyBERT()
+        print("Warning: Using default KeyBERT model instead of PubMedBERT")
 
     return {
         "led": {