caubetotbunggg commited on
Commit
d46f72b
·
verified ·
1 Parent(s): e208d53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from sentence_transformers import SentenceTransformer
3
 
4
  # Load model BGEM3
5
- model = SentenceTransformer("BAAI/bge-m3")
6
 
7
  def embed_text(text):
8
  if not text.strip():
@@ -12,7 +12,7 @@ def embed_text(text):
12
 
13
  # Gradio UI
14
  with gr.Blocks() as demo:
15
- gr.Markdown("## 🔎 BGEM3 Embedding Demo")
16
  inp = gr.Textbox(label="Nhập câu hoặc đoạn văn")
17
  out = gr.JSON(label="Vector (10 giá trị đầu)")
18
  btn = gr.Button("Generate Embedding")
 
2
  from sentence_transformers import SentenceTransformer
3
 
4
  # Load model BGEM3
5
+ model = SentenceTransformer("intfloat/multilingual-e5-base")
6
 
7
  def embed_text(text):
8
  if not text.strip():
 
12
 
13
  # Gradio UI
14
  with gr.Blocks() as demo:
15
+ gr.Markdown("## 🔎 e5 Embedding Demo")
16
  inp = gr.Textbox(label="Nhập câu hoặc đoạn văn")
17
  out = gr.JSON(label="Vector (10 giá trị đầu)")
18
  btn = gr.Button("Generate Embedding")