VyLala commited on
Commit
756ed18
·
verified ·
1 Parent(s): fb8d818

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -1
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 = "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"]
 
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"]