feat: update model

app.py

@@ -1,7 +1,25 @@
 import gradio as gr
+from sentence_transformers import SentenceTransformer
 
-def greet(name):
-    return "Hello " + name + "!!"
+# Tải mô hình
+model = SentenceTransformer('Alibaba-NLP/gte-multilingual-base')
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
+def gte_model(sentences: list):
+    try:
+        # Mã hóa các câu
+        embeddings = model.encode(sentences)
+        return embeddings.tolist()  # Chuyển đổi numpy array sang danh sách
+    except Exception as e:
+        return f"Error: {str(e)}"
+
+# Tạo giao diện Gradio
+demo = gr.Interface(
+    fn=gte_model,
+    inputs=gr.inputs.Textbox(lines=5, placeholder="Nhập các câu ở đây, mỗi câu trên một dòng..."),
+    outputs=gr.outputs.JSON(label="Kết quả mã hóa"),
+    title="Mô hình GTE Multilingual",
+    description="Nhập các câu để nhận mã hóa từ mô hình GTE Multilingual. Kết quả sẽ được trả về dưới dạng danh sách mã hóa."
+)
+
+# Khởi chạy giao diện
+demo.launch()

requirements.txt

@@ -17,10 +17,25 @@ httpx==0.27.2
 huggingface-hub==0.26.2
 idna==3.10
 Jinja2==3.1.4
+joblib==1.4.2
 markdown-it-py==3.0.0
 MarkupSafe==2.1.5
 mdurl==0.1.2
+mpmath==1.3.0
+networkx==3.4.2
 numpy==2.1.3
+nvidia-cublas-cu12==12.4.5.8
+nvidia-cuda-cupti-cu12==12.4.127
+nvidia-cuda-nvrtc-cu12==12.4.127
+nvidia-cuda-runtime-cu12==12.4.127
+nvidia-cudnn-cu12==9.1.0.70
+nvidia-cufft-cu12==11.2.1.3
+nvidia-curand-cu12==10.3.5.147
+nvidia-cusolver-cu12==11.6.1.9
+nvidia-cusparse-cu12==12.3.1.170
+nvidia-nccl-cu12==2.21.5
+nvidia-nvjitlink-cu12==12.4.127
+nvidia-nvtx-cu12==12.4.127
 orjson==3.10.11
 packaging==24.1
 pandas==2.2.3
@@ -33,17 +48,28 @@ python-dateutil==2.9.0.post0
 python-multipart==0.0.12
 pytz==2024.2
 PyYAML==6.0.2
+regex==2024.9.11
 requests==2.32.3
 rich==13.9.4
 ruff==0.7.2
 safehttpx==0.1.1
+safetensors==0.4.5
+scikit-learn==1.5.2
+scipy==1.14.1
 semantic-version==2.10.0
+sentence-transformers==3.2.1
 shellingham==1.5.4
 six==1.16.0
 sniffio==1.3.1
 starlette==0.41.2
+sympy==1.13.1
+threadpoolctl==3.5.0
+tokenizers==0.20.2
 tomlkit==0.12.0
+torch==2.5.1
 tqdm==4.66.6
+transformers==4.46.1
+triton==3.1.0
 typer==0.12.5
 typing_extensions==4.12.2
 tzdata==2024.2