acherkrau commited on
Commit
ae4544d
·
verified ·
1 Parent(s): 0e8acca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,7 +6,7 @@ import os
6
  from sentence_transformers import SentenceTransformer
7
 
8
  model = SentenceTransformer(
9
- "dunzhang/stella_en_400M_v5",
10
  trust_remote_code=True,
11
  device="cpu",
12
  config_kwargs={"use_memory_efficient_attention": False, "unpad_inputs": False}
@@ -45,7 +45,7 @@ def parse(data):
45
  async def get_embedding(embedding: Embedding, req: Request):
46
 
47
  token = req.headers.get("Authorization")
48
- if os.environ.get('token') != token[7:]:
49
  raise HTTPException(status_code=401, detail="Unauthorized.")
50
 
51
  if model == None:
 
6
  from sentence_transformers import SentenceTransformer
7
 
8
  model = SentenceTransformer(
9
+ "dunzhang/stella_en_1.5B_v5",
10
  trust_remote_code=True,
11
  device="cpu",
12
  config_kwargs={"use_memory_efficient_attention": False, "unpad_inputs": False}
 
45
  async def get_embedding(embedding: Embedding, req: Request):
46
 
47
  token = req.headers.get("Authorization")
48
+ if not token or os.environ.get('token') != token[7:]:
49
  raise HTTPException(status_code=401, detail="Unauthorized.")
50
 
51
  if model == None: