waseoke commited on
Commit
d9888f2
·
verified ·
1 Parent(s): de8ce12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -7,6 +7,7 @@ from torch.nn import Embedding
7
  client = MongoClient("mongodb+srv://waseoke:[email protected]/test?retryWrites=true&w=majority&tls=true&tlsAllowInvalidCertificates=true")
8
  db = client["패션"]
9
  product_collection = db["여성의류"]
 
10
 
11
  # Hugging Face의 한국어 BERT 모델 및 토크나이저 로드 (예: klue/bert-base)
12
  tokenizer = BertTokenizer.from_pretrained("klue/bert-base")
 
7
  client = MongoClient("mongodb+srv://waseoke:[email protected]/test?retryWrites=true&w=majority&tls=true&tlsAllowInvalidCertificates=true")
8
  db = client["패션"]
9
  product_collection = db["여성의류"]
10
+ embedding_collection = db["product_embeddings"] # 임베딩을 저장할 컬렉션
11
 
12
  # Hugging Face의 한국어 BERT 모델 및 토크나이저 로드 (예: klue/bert-base)
13
  tokenizer = BertTokenizer.from_pretrained("klue/bert-base")