Spaces:
Sleeping
Sleeping
changes in encode
Browse files
app.py
CHANGED
@@ -9,11 +9,13 @@ from sentence_transformers import SentenceTransformer, CrossEncoder
|
|
9 |
from rank_bm25 import BM25Okapi
|
10 |
import gradio as gr
|
11 |
|
|
|
|
|
12 |
# β
Load Models
|
13 |
spacy.cli.download("en_core_web_sm")
|
14 |
nlp = spacy.load("en_core_web_sm")
|
15 |
embed_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
|
16 |
-
cross_encoder = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-
|
17 |
|
18 |
# β
Load API Key from Hugging Face Secrets
|
19 |
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|
|
|
9 |
from rank_bm25 import BM25Okapi
|
10 |
import gradio as gr
|
11 |
|
12 |
+
print(CrossEncoder.list_pretrained_models())
|
13 |
+
|
14 |
# β
Load Models
|
15 |
spacy.cli.download("en_core_web_sm")
|
16 |
nlp = spacy.load("en_core_web_sm")
|
17 |
embed_model = SentenceTransformer("sentence-transformers/all-MiniLM-L6-v2")
|
18 |
+
cross_encoder = CrossEncoder("cross-encoder/ms-marco-MiniLM-L-12-v2")
|
19 |
|
20 |
# β
Load API Key from Hugging Face Secrets
|
21 |
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|