Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -6,7 +6,7 @@ import os
|
|
| 6 |
from sentence_transformers import SentenceTransformer
|
| 7 |
|
| 8 |
model = SentenceTransformer(
|
| 9 |
-
"dunzhang/
|
| 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:
|