VietCat commited on
Commit
c4fe2d6
·
1 Parent(s): 2c901bd

update app logic

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def embed():
20
  # Lấy embedding từ hidden state đầu tiên
21
  embedding = outputs.last_hidden_state[:, 0, :].numpy().tolist() # Dùng .numpy() để chuyển từ TensorFlow tensor sang list
22
 
23
- return jsonify({"embedding": embedding})
24
 
25
  @app.route('/', methods=['GET'])
26
  def index():
 
20
  # Lấy embedding từ hidden state đầu tiên
21
  embedding = outputs.last_hidden_state[:, 0, :].numpy().tolist() # Dùng .numpy() để chuyển từ TensorFlow tensor sang list
22
 
23
+ return jsonify({"embeddings": embedding})
24
 
25
  @app.route('/', methods=['GET'])
26
  def index():