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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -4
app.py CHANGED
@@ -81,16 +81,19 @@ def load_resources():
81
  raise
82
 
83
  from functools import lru_cache
84
- @lru_cache(maxsize=1000)
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
- 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)
@@ -102,8 +105,9 @@ def predict(vector):
102
 
103
  except Exception as e:
104
  # 添加详细错误日志
105
- import traceback
106
- error_msg = f"处理错误: {str(e)}\n{traceback.format_exc()}"
 
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