Update app.py
Browse files
app.py
CHANGED
@@ -81,16 +81,19 @@ def load_resources():
|
|
81 |
raise
|
82 |
|
83 |
from functools import lru_cache
|
84 |
-
@lru_cache(maxsize=
|
85 |
def predict(vector):
|
86 |
"""处理768维向量输入并返回答案"""
|
87 |
print(f"输入向量维度: {np.array(vector).shape}")
|
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 |
-
|
|
|
|
|
94 |
|
95 |
# 添加实际处理逻辑
|
96 |
D, I = index.search(np.array(vector, dtype=np.float32), k=3)
|
@@ -102,8 +105,9 @@ def predict(vector):
|
|
102 |
|
103 |
except Exception as e:
|
104 |
# 添加详细错误日志
|
105 |
-
|
106 |
-
|
|
|
107 |
print(error_msg)
|
108 |
return error_msg
|
109 |
|
|
|
81 |
raise
|
82 |
|
83 |
from functools import lru_cache
|
84 |
+
@lru_cache(maxsize=100)
|
85 |
def predict(vector):
|
86 |
"""处理768维向量输入并返回答案"""
|
87 |
print(f"输入向量维度: {np.array(vector).shape}")
|
88 |
try:
|
89 |
if not isinstance(vector, list) or len(vector) == 0:
|
90 |
+
print("错误:输入格式无效")
|
91 |
return "错误:输入格式无效"
|
92 |
|
93 |
if len(vector) != 1 or len(vector[0]) != 768:
|
94 |
+
error_msg = f"错误:需要1x768的二维数组,收到{len(vector)}x{len(vector[0]) if vector else 0}"
|
95 |
+
print(error_msg)
|
96 |
+
return error_msg
|
97 |
|
98 |
# 添加实际处理逻辑
|
99 |
D, I = index.search(np.array(vector, dtype=np.float32), k=3)
|
|
|
105 |
|
106 |
except Exception as e:
|
107 |
# 添加详细错误日志
|
108 |
+
error_msg = "处理错误,请重试或联系管理员"
|
109 |
+
# import traceback
|
110 |
+
# error_msg = f"处理错误: {str(e)}\n{traceback.format_exc()}"
|
111 |
print(error_msg)
|
112 |
return error_msg
|
113 |
|