Spaces:
Sleeping
Sleeping
File size: 596 Bytes
4f70a94 0266b6d 36a5a1e 0266b6d d3f7acb 0266b6d 597075b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import streamlit as st
from sentence_transformers import SentenceTransformer
import os
os.environ["SENTENCE_TRANSFORMERS_DISABLE_ONNX"] = "1"
os.environ["TRANSFORMERS_CACHE"] = "/tmp/.cache"
os.environ["HF_HOME"] = "/tmp/.cache"
os.makedirs("/tmp/.cache", exist_ok=True)
model = SentenceTransformer(
"sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2",
cache_folder="/tmp/.cache"
)
def generate_embedding(text):
if not text or not isinstance(text, str) or text.strip() == "":
raise ValueError("Input text is empty or invalid.")
return model.encode(text)
|