IotaCluster commited on
Commit
ac5cb04
·
verified ·
1 Parent(s): fd20dab

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -6,8 +6,8 @@ import torch
6
 
7
  # 1. Dense embedding model (HF bi-encoder)
8
  # dense_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
9
- dense_model = SentenceTransformer('distiluse-base-multilingual-cased-v2')
10
-
11
  def embed_dense(text: str):
12
  if not text.strip():
13
  return {"error": "Input text is empty."}
 
6
 
7
  # 1. Dense embedding model (HF bi-encoder)
8
  # dense_model = SentenceTransformer('sentence-transformers/all-MiniLM-L6-v2')
9
+ # dense_model = SentenceTransformer('distiluse-base-multilingual-cased-v2')
10
+ dense_model = SentenceTransformer('multi-qa-mpnet-base-cos-v1')
11
  def embed_dense(text: str):
12
  if not text.strip():
13
  return {"error": "Input text is empty."}