GOGO198 commited on
Commit
e2a605c
·
verified ·
1 Parent(s): aa0b718

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -88,10 +88,9 @@ def predict(vector):
88
  try:
89
  if not isinstance(vector, list) or len(vector) == 0:
90
  return "错误:输入格式无效"
91
-
92
- # 确保输入数据完整
93
- if len(vector[0]) != 768:
94
- return "维度错误:需要768维向量"
95
 
96
  # 添加实际处理逻辑
97
  D, I = index.search(np.array(vector, dtype=np.float32), k=3)
 
88
  try:
89
  if not isinstance(vector, list) or len(vector) == 0:
90
  return "错误:输入格式无效"
91
+
92
+ if len(vector) != 1 or len(vector[0]) != 768:
93
+ return f"错误:需要1x768的二维数组,收到{len(vector)}x{len(vector[0]) if vector else 0}"
 
94
 
95
  # 添加实际处理逻辑
96
  D, I = index.search(np.array(vector, dtype=np.float32), k=3)