akashjayampu commited on
Commit
f0db406
·
verified ·
1 Parent(s): 8498ff5

Update src/embedding.py

Browse files
Files changed (1) hide show
  1. src/embedding.py +1 -3
src/embedding.py CHANGED
@@ -1,11 +1,9 @@
1
- # embedding.py
2
  from sentence_transformers import SentenceTransformer
3
  import streamlit as st
4
 
5
  @st.cache_resource
6
  def get_embed_model():
7
- # Use safetensors if supported (newer versions of sentence-transformers)
8
- return SentenceTransformer("all-MiniLM-L6-v2")
9
 
10
  def get_embeddings(texts):
11
  model = get_embed_model()
 
 
1
  from sentence_transformers import SentenceTransformer
2
  import streamlit as st
3
 
4
  @st.cache_resource
5
  def get_embed_model():
6
+ return SentenceTransformer("all-MiniLM-L6-v2")
 
7
 
8
  def get_embeddings(texts):
9
  model = get_embed_model()