jxtan commited on
Commit
404b09b
·
verified ·
1 Parent(s): ad6d176

Update sentence_embeddings.py

Browse files
Files changed (1) hide show
  1. sentence_embeddings.py +1 -1
sentence_embeddings.py CHANGED
@@ -5,7 +5,7 @@ from transformers import AutoTokenizer, AutoModel
5
  import torch
6
  from datetime import datetime
7
  from logger import log
8
- from hf_to_api.config import TEST_MODE
9
 
10
  router = APIRouter()
11
 
 
5
  import torch
6
  from datetime import datetime
7
  from logger import log
8
+ from config import TEST_MODE
9
 
10
  router = APIRouter()
11