added code and updated for BAAI/bge-small-en-v1.5 model

Dockerfile

@@ -1,5 +1,5 @@
 # Define global args
-ARG MODEL="BAAI/bge-large-en"
+ARG MODEL="BAAI/bge-small-en-v1.5"
 
 FROM debian:bullseye-slim AS build-image
 

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.

README.md

@@ -1,19 +1,19 @@
 ---
-title: BAAI/bge-large-en OpenAI API-Compatible Endpoint
+title: BAAI/bge-small-en-v1.5 OpenAI API-Compatible Endpoint
 emoji: 👁
 colorFrom: gray
 colorTo: pink
 sdk: docker
 models:
-  - BAAI/bge-large-en
+  - BAAI/bge-small-en-v1.5
 tags:
   - inference api
   - openai-api compatible
   - open-text-embeddings
-  - bge-large-en
+  - bge-small-en-v1.5
 pinned: false
 ---
 
-# BAAI/bge-large-en OpenAI API-Compatible Endpoint
+# BAAI/bge-small-en-v1.5 OpenAI API-Compatible Endpoint
 
-Please refer to the [main screen](https://huggingface.co/spaces/limcheekin/bge-large-en) for more information.
+Please refer to the [main screen](https://huggingface.co/spaces/limcheekin/bge-small-en-v1.5) for more information.

main.py

@@ -0,0 +1,21 @@
+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()
+
+
+@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

@@ -1,3 +1,3 @@
 #!/bin/sh
 
-python -B -m open.text.embeddings.server
+python -B main.py