vladbogo commited on
Commit
e56b403
1 Parent(s): cbe0c63

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. run_pipeline.py +9 -10
run_pipeline.py CHANGED
@@ -105,17 +105,16 @@ def get_fact_checked(text_input, model="gpt-3.5-turbo", mode="slow"):
105
 
106
  # STEP 8
107
  print("Step8: Formatting")
108
- if mode == "slow":
109
- step8 = FormatDocument(model=model, footnote_style="verbose")
110
- step8_md = step8.reformat_document_to_include_claims(
111
- text_input, step7_json, footnote_style="verbose"
112
- )
113
- step8_md_terse = step8.reformat_document_to_include_claims(
114
- text_input, step7_json, footnote_style="terse"
115
- )
116
 
117
- results["fact_checked_md"] = copy.deepcopy(step8_md)
118
- results["fact_checked_terse"] = copy.deepcopy(step8_md_terse)
119
  return results
120
 
121
 
 
105
 
106
  # STEP 8
107
  print("Step8: Formatting")
108
+ step8 = FormatDocument(model=model, footnote_style="verbose")
109
+ step8_md = step8.reformat_document_to_include_claims(
110
+ text_input, step7_json, footnote_style="verbose"
111
+ )
112
+ step8_md_terse = step8.reformat_document_to_include_claims(
113
+ text_input, step7_json, footnote_style="terse"
114
+ )
 
115
 
116
+ results["fact_checked_md"] = copy.deepcopy(step8_md)
117
+ results["fact_checked_terse"] = copy.deepcopy(step8_md_terse)
118
  return results
119
 
120