Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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")
|