alfa95 commited on
Commit
dc800a5
Β·
1 Parent(s): ad9c60b

changes in encode

Browse files
Files changed (1) hide show
  1. app.py +3 -1
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-6-v2")
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")