B4869 commited on
Commit
d495ad2
·
1 Parent(s): 8379ff4

update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -1
app.py CHANGED
@@ -17,7 +17,20 @@ model = SentenceTransformer("google/embeddinggemma-300m")
17
 
18
  @app.route("/", methods=['GET'])
19
  def get_start():
20
- return jsonify({"message": "Server is running"})
 
 
 
 
 
 
 
 
 
 
 
 
 
21
 
22
  @app.route("/api/v1/embeddings", methods=['POST'])
23
  def return_query_embedding():
 
17
 
18
  @app.route("/", methods=['GET'])
19
  def get_start():
20
+ # 1. รับ JSON Payload
21
+ try:
22
+ query = "Hello test."
23
+ query_embeddings = model.encode_query(query)
24
+ query_embeddings = query_embeddings.tolist()
25
+ query_embeddings_json = {
26
+ "query_embeddings": query_embeddings
27
+ }
28
+ return jsonify(query_embeddings_json)
29
+ except Exception as e:
30
+ # ดักจับกรณีที่ JSON ที่ส่งมาผิดรูปแบบ
31
+ return jsonify({"error": str(e)}), 500
32
+
33
+
34
 
35
  @app.route("/api/v1/embeddings", methods=['POST'])
36
  def return_query_embedding():