Spaces:
Build error
Build error
Update logical_reasoning_utils.py
Browse files
llm_toolkit/logical_reasoning_utils.py
CHANGED
@@ -116,9 +116,13 @@ P2_few_shot = """你是一个情景猜谜游戏的主持人。游戏规则如下
|
|
116 |
回答:
|
117 |
"""
|
118 |
|
119 |
-
|
120 |
-
|
121 |
-
translation_dict =
|
|
|
|
|
|
|
|
|
122 |
|
123 |
def get_prompt_template(using_p1=True, chinese_prompt=True):
|
124 |
if using_p1:
|
|
|
116 |
回答:
|
117 |
"""
|
118 |
|
119 |
+
try:
|
120 |
+
df_translation = pd.read_csv("datasets/mgtv/unique_translations.csv")
|
121 |
+
translation_dict = df_translation.set_index("chinese").to_dict(orient="index")
|
122 |
+
translation_dict = {k: v["english"] for k, v in translation_dict.items()}
|
123 |
+
except Exception as e:
|
124 |
+
print(e)
|
125 |
+
translation_dict = {}
|
126 |
|
127 |
def get_prompt_template(using_p1=True, chinese_prompt=True):
|
128 |
if using_p1:
|