Spaces:
Running
Running
Update model.py
Browse files
model.py
CHANGED
@@ -1263,7 +1263,7 @@ def query_document_info(query_word, alternative_query_word, metadata, master_str
|
|
1263 |
# extracted_type = merge_metadata["sample_type"]
|
1264 |
# sample_type_explanation = merge_metadata["sample_type_explanation"]
|
1265 |
# for the output_format that is not default
|
1266 |
-
if output_format_str
|
1267 |
outputs = output_format_str.split(", ")
|
1268 |
extracted_country, extracted_type = metadata_list[outputs[0]]["answer"], metadata_list[outputs[1]]["answer"]
|
1269 |
country_explanation,sample_type_explanation = metadata_list[outputs[0]][outputs[0]+"_explanation"], metadata_list[outputs[1]][outputs[1]+"_explanation"]
|
|
|
1263 |
# extracted_type = merge_metadata["sample_type"]
|
1264 |
# sample_type_explanation = merge_metadata["sample_type_explanation"]
|
1265 |
# for the output_format that is not default
|
1266 |
+
if output_format_str == "country_name, modern/ancient/unknown":
|
1267 |
outputs = output_format_str.split(", ")
|
1268 |
extracted_country, extracted_type = metadata_list[outputs[0]]["answer"], metadata_list[outputs[1]]["answer"]
|
1269 |
country_explanation,sample_type_explanation = metadata_list[outputs[0]][outputs[0]+"_explanation"], metadata_list[outputs[1]][outputs[1]+"_explanation"]
|