init

.gitignore

@@ -0,0 +1 @@
+.idea/

Dockerfile

@@ -0,0 +1,16 @@
+# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
+# you will also find guides on how best to write your Dockerfile
+
+FROM python:3.11
+
+RUN useradd -m -u 1000 user
+USER user
+ENV PATH="/home/user/.local/bin:$PATH"
+
+WORKDIR /app
+
+COPY --chown=user ./requirements.txt requirements.txt
+RUN pip install --no-cache-dir --upgrade -r requirements.txt
+
+COPY --chown=user . /app
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]

app.py

@@ -0,0 +1,24 @@
+from flask import Flask, request, jsonify
+from sentence_transformers import SentenceTransformer
+import logging
+
+logging.basicConfig(level=logging.DEBUG)
+
+app = Flask(__name__)
+
+# Load the Qwen3-Embedding-0.6B model (adjust the path as needed)
+model = SentenceTransformer(model_name_or_path="Qwen/Qwen3-Embedding-0.6B")
+
+@app.route('/embed', methods=['POST'])
+def get_embedding():
+    try:
+        text = request.json.get('text', '')
+        if not text:
+            return jsonify({"error": "No text provided"}), 400
+        
+        # Generate embeddings for the input text
+        document_embeddings = model.encode(text)
+        arr_list = document_embeddings.tolist()
+        return jsonify({"embedding": arr_list})
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500

requirements.txt

@@ -0,0 +1,4 @@
+flask
+sentence-transformers
+transformers
+torch