moraxgiga commited on
Commit
150b3dc
·
verified ·
1 Parent(s): 214e49e

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -3,9 +3,10 @@ from pydantic import BaseModel, root_validator
3
  from transformers import AutoModel
4
  from typing import List
5
  import os, platform
 
6
 
7
 
8
- model = AutoModel.from_pretrained('jinaai/jina-embeddings-v2-base-en',token = "hf_GkUomApayMBJteRvrjvslfyLRvfp QRckba".replace(" ", ""), trust_remote_code=True)
9
 
10
  app = FastAPI()
11
 
 
3
  from transformers import AutoModel
4
  from typing import List
5
  import os, platform
6
+ !huggingface-cli login --token os.environ.get("HF_TOKEN")
7
 
8
 
9
+ model = AutoModel.from_pretrained('jinaai/jina-embeddings-v2-base-en', trust_remote_code=True)
10
 
11
  app = FastAPI()
12