Vu Minh Chien commited on
Commit
cc741b8
·
1 Parent(s): 3197697
services/sentence_transformer_service.py CHANGED
@@ -215,7 +215,7 @@ class SentenceTransformerService:
215
  # Load name map data
216
  name_map_file = os.path.join(DATA_DIR, "nameMapData.csv")
217
  if os.path.exists(name_map_file):
218
- self.df_name_map_data = pd.read_csv(name_map_file)
219
  print(f"Loaded name map data: {len(self.df_name_map_data)} entries")
220
 
221
  # Load sub subject and name map data
@@ -233,7 +233,7 @@ class SentenceTransformerService:
233
  # Load abstract map data
234
  abstract_map_file = os.path.join(DATA_DIR, "abstractMapData.csv")
235
  if os.path.exists(abstract_map_file):
236
- self.df_abstract_map_data = pd.read_csv(abstract_map_file)
237
  print(
238
  f"Loaded abstract map data: {len(self.df_abstract_map_data)} entries"
239
  )
 
215
  # Load name map data
216
  name_map_file = os.path.join(DATA_DIR, "nameMapData.csv")
217
  if os.path.exists(name_map_file):
218
+ self.df_name_map_data = pd.read_csv(name_map_file, dtype={'外部・内部設定タイプ': str})
219
  print(f"Loaded name map data: {len(self.df_name_map_data)} entries")
220
 
221
  # Load sub subject and name map data
 
233
  # Load abstract map data
234
  abstract_map_file = os.path.join(DATA_DIR, "abstractMapData.csv")
235
  if os.path.exists(abstract_map_file):
236
+ self.df_abstract_map_data = pd.read_csv(abstract_map_file, dtype={'摘要タイプ': str})
237
  print(
238
  f"Loaded abstract map data: {len(self.df_abstract_map_data)} entries"
239
  )