Update app.py

app.py

@@ -50,12 +50,12 @@ def get_st(tgt):
         st.error("Failed to obtain service ticket from UMLS API.")
         st.stop()
 
-# Text embedding
+# Text embedding (tokenizer and model are unhashable, prefix with underscore)
 @st.cache_resource
-def embed_text(text, tokenizer, model):
-    inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True)
+def embed_text(text, _tokenizer, _model):
+    inputs = _tokenizer(text, return_tensors="pt", truncation=True, padding=True)
     with torch.no_grad():
-        outputs = model(**inputs)
+        outputs = _model(**inputs)
     emb = outputs.last_hidden_state[:, 0, :].squeeze().cpu().numpy()
     return emb / np.linalg.norm(emb)