Spaces:
Running
Running
Vu Minh Chien
commited on
Commit
·
cc741b8
1
Parent(s):
3197697
fix dtype
Browse files
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 |
)
|