Update app.py
Browse files
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 |
-
|
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)
|