import gradio as gr
from sentence_transformers import SentenceTransformer

# 🔥 替换为官方公开模型（解决401权限错误）
model = SentenceTransformer("codefuse-ai/F2LLM-v2-0.6B")

def get_embedding(text):
    embedding = model.encode(text)
    return {
        "输入文本": text,
        "向量预览（前10维）": embedding[:10].tolist(),
        "向量总维度": len(embedding)
    }

demo = gr.Interface(
    fn=get_embedding,
    inputs=gr.Textbox(label="输入文本", placeholder="请输入要生成向量的文本"),
    outputs=gr.JSON(label="生成结果"),
    title="F2LLM-v2 文本嵌入演示",
    description="基于 F2LLM-v2-0.6B 的多语种文本嵌入模型"
)

if __name__ == "__main__":
    demo.launch()