Jindra Snobl commited on
Commit
81d4a71
·
1 Parent(s): 7a84c6b

Switched model to public one

Browse files
Files changed (1) hide show
  1. api/embed.py +6 -3
api/embed.py CHANGED
@@ -11,10 +11,13 @@ huggingface_token = os.environ.get("HF_TOKEN", "")
11
  print(huggingface_token)
12
 
13
  # login(huggingface_token)
 
 
 
 
 
 
14
 
15
- auto_tokenizer = AutoTokenizer.from_pretrained(
16
- "CocoonBusiness/VectorSearch", token=huggingface_token, low_cpu_mem_usage=True
17
- )
18
  model = AutoModel.from_pretrained(
19
  "jegorkitskerkin/robbert-v2-dutch-base-mqa-finetuned", low_cpu_mem_usage=True
20
  )
 
11
  print(huggingface_token)
12
 
13
  # login(huggingface_token)
14
+ # Model is private?
15
+ # auto_tokenizer = AutoTokenizer.from_pretrained(
16
+ # "CocoonBusiness/VectorSearch", token=huggingface_token, low_cpu_mem_usage=True
17
+ # )
18
+
19
+ auto_tokenizer = AutoTokenizer.from_pretrained("xValentim/vector-search-bert-based", low_cpu_mem_usage=True)
20
 
 
 
 
21
  model = AutoModel.from_pretrained(
22
  "jegorkitskerkin/robbert-v2-dutch-base-mqa-finetuned", low_cpu_mem_usage=True
23
  )