Change model to sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2 d-384

Dockerfile

@@ -0,0 +1,46 @@
+# Cloned from https://huggingface.co/spaces/limcheekin/bge-small-en-v1.5/tree/main
+# Define global args
+# ARG MODEL="BAAI/bge-small-en-v1.5"
+ARG MODEL="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2"
+
+FROM debian:bullseye-slim AS build-image
+
+# Include global args in this stage of the build
+ARG MODEL
+ENV MODEL=${MODEL}
+
+COPY ./download.sh ./
+
+# Install build dependencies
+RUN apt-get update && \
+    apt-get install -y git-lfs
+
+RUN chmod +x *.sh && \
+    ./download.sh && \
+    rm *.sh
+
+# Stage 3 - final runtime image
+# Grab a fresh copy of the Python image
+FROM python:3.11-slim
+
+# Include global args in this stage of the build
+ARG MODEL
+ENV MODEL=${MODEL}
+ENV NORMALIZE_EMBEDDINGS=1
+ENV HF_HOME="/tmp/hf_home"
+# Set environment variable for the host
+ENV HOST=0.0.0.0
+ENV PORT=7860
+
+COPY --from=build-image ${MODEL} ${MODEL}
+COPY ./main.py ./
+COPY ./start_server.sh ./
+# COPY ./index.html ./
+RUN pip install --no-cache-dir open-text-embeddings[server] && \
+    chmod +x ./start_server.sh
+
+# Expose a port for the server
+EXPOSE ${PORT}
+
+# Run the server start script
+CMD ["/bin/sh", "./start_server.sh"]

LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2023 Lim Chee Kin
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

download.sh

@@ -0,0 +1,9 @@
+mkdir -p $MODEL
+git lfs install --skip-smudge
+git clone https://huggingface.co/$MODEL $MODEL
+cd $MODEL
+git lfs pull
+git lfs install --force
+rm -rf .git
+pwd
+ls -l

main.py

@@ -0,0 +1,35 @@
+from open.text.embeddings.server.app import create_app
+from fastapi.responses import HTMLResponse
+import os
+
+app = create_app()
+
+# Read the content of index.html once and store it in memory
+# with open("index.html", "r") as f:     content = f.read()
+content = """
+<ul>
+      <li>
+        The API endpoint:
+        <a href="https://mikeee-emb384-oai.hf.space/v1"
+          >https://mikeee-emb384-oai.hf.space/v1</a
+        >
+      </li>
+      <li>
+        The API doc:
+        <a href="https://mikeee-emb384-oai.hf.space/docs"
+          >https://mikeee-emb384-oai.hf.space/docs</a
+        >
+      </li>
+    </ul>
+"""
+
+@app.get("/", response_class=HTMLResponse)
+async def read_items():
+    return content
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app,
+                host=os.environ["HOST"],
+                port=int(os.environ["PORT"])
+                )

start_server.sh

@@ -0,0 +1,3 @@
+#!/bin/sh
+
+python -B main.py