Spaces:

Chatbot-TLU
/

M_chatbot

Sleeping

minh-4T commited on Mar 29

Commit

8723e94

verified ·

1 Parent(s): 8193b5a

Update core/models.py

Files changed (1) hide show

core/models.py CHANGED Viewed

@@ -7,5 +7,6 @@ embeddings = HuggingFaceEmbeddings(
     model_kwargs={'device': 'cpu'},
     encode_kwargs={'normalize_embeddings': True}
 )
 llm = None

     model_kwargs={'device': 'cpu'},
     encode_kwargs={'normalize_embeddings': True}
 )
+cross_encoder = CrossEncoder(CROSS_ENCODER_MODEL, device='cpu')
 llm = None