import gradio as gr
from sentence_transformers import SentenceTransformer

# 모델 로드
model = SentenceTransformer("sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2")

# 문장을 입력받아 임베딩 생성
def embed_text(text):
    embedding = model.encode(text).tolist()
    return embedding

# Gradio 인터페이스 설정
demo = gr.Interface(fn=embed_text, inputs="text", outputs="json", title="Text Embedding API")

# 실행
if __name__ == "__main__":
    demo.launch()