Spaces:

ibombonato
/

Semantic-search-br

Sleeping

ibombonato commited on Jul 9

Commit

82fdd67

verified ·

1 Parent(s): d44b7ca

Upload folder using huggingface_hub

Files changed (1) hide show

app.py CHANGED Viewed

@@ -9,7 +9,8 @@ import numpy
 # --- 1. SETUP MODELS AND DATABASE ---
 print("Loading embedding model...")
-embedding_model = SentenceTransformer('rufimelo/bert-large-portuguese-cased-sts')
 client = chromadb.Client()
 collection = client.get_or_create_collection(

 # --- 1. SETUP MODELS AND DATABASE ---
 print("Loading embedding model...")
+#embedding_model = SentenceTransformer('rufimelo/bert-large-portuguese-cased-sts')
+embedding_model = SentenceTransformer('jmbrito/ptbr-similarity-e5-small')
 client = chromadb.Client()
 collection = client.get_or_create_collection(