File size: 816 Bytes
56f89b6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 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 |