Update app.py

app.py

@@ -1,24 +1,28 @@
-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
+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
+
+
+if __name__ == '__main__':
+    app.run(host='0.0.0.0', port=7860)