parkerjj commited on
Commit
fbe1529
·
1 Parent(s): 0739750

优化预测函数的缓存检查逻辑,确保输入文本不为空时才进行缓存查找

Browse files
Files changed (1) hide show
  1. blkeras.py +1 -1
blkeras.py CHANGED
@@ -148,7 +148,7 @@ def predict(text: str, stock_codes: list):
148
  # 生成唯一键值
149
  cache_key = generate_key(lemmatized_entry)
150
  # 检查缓存中是否已有结果
151
- if cache_key in prediction_cache:
152
  print(f"Cache hit: {cache_key}" )
153
  return prediction_cache[cache_key]
154
 
 
148
  # 生成唯一键值
149
  cache_key = generate_key(lemmatized_entry)
150
  # 检查缓存中是否已有结果
151
+ if input_text != "EMPTY_TEXT" and cache_key in prediction_cache:
152
  print(f"Cache hit: {cache_key}" )
153
  return prediction_cache[cache_key]
154