Blaise-g commited on
Commit
09b0aa0
Β·
1 Parent(s): b59f31f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -72,18 +72,18 @@ def proc_submission(
72
  msg = None
73
 
74
  if (model_type == "LED" and summary_type == "Detailed"):
75
- model = model_led_det
76
  elif (model_type == "LED" and summary_type == "TLDR"):
77
- model = model_led_tldr
78
  elif (model_type == "LongT5" and summary_type == "Detailed"):
79
- model = model_det
80
  else:
81
- model = model_tldr
82
  #tokenizer_led_det if (model_type == "LED" and summary_type == "Detailed") else tokenizer_det,
83
  _summaries = summarize_via_tokenbatches(
84
  tr_in,
85
  tokenizer_led_det if (model_type == "LED") else tokenizer_det,
86
- model = model,
87
  batch_length=token_batch_length,
88
  **settings,
89
  )
@@ -155,8 +155,8 @@ if __name__ == "__main__":
155
 
156
  model_det, tokenizer_det = load_model_and_tokenizer("Blaise-g/longt5_tglobal_large_sumpubmed")
157
  model_tldr, tokenizer_tldr = load_model_and_tokenizer("Blaise-g/longt5_tglobal_large_scitldr")
158
- model_led_det, tokenizer_led_det = load_model_and_tokenizer("Blaise-g/led_pubmed_sumpubmed_1")
159
- model_led_tldr, tokenizer_led_tldr = load_model_and_tokenizer("Blaise-g/led_large_sumpbumed_scitldr")
160
 
161
  name_to_path = load_example_filenames(_here / "examples")
162
  logging.info(f"Loaded {len(name_to_path)} examples")
 
72
  msg = None
73
 
74
  if (model_type == "LED" and summary_type == "Detailed"):
75
+ m = model_led_det
76
  elif (model_type == "LED" and summary_type == "TLDR"):
77
+ m = model_led_tldr
78
  elif (model_type == "LongT5" and summary_type == "Detailed"):
79
+ m = model_det
80
  else:
81
+ m = model_tldr
82
  #tokenizer_led_det if (model_type == "LED" and summary_type == "Detailed") else tokenizer_det,
83
  _summaries = summarize_via_tokenbatches(
84
  tr_in,
85
  tokenizer_led_det if (model_type == "LED") else tokenizer_det,
86
+ model = m,
87
  batch_length=token_batch_length,
88
  **settings,
89
  )
 
155
 
156
  model_det, tokenizer_det = load_model_and_tokenizer("Blaise-g/longt5_tglobal_large_sumpubmed")
157
  model_tldr, tokenizer_tldr = load_model_and_tokenizer("Blaise-g/longt5_tglobal_large_scitldr")
158
+ #model_led_det, tokenizer_led_det = #load_model_and_tokenizer("Blaise-g/led_pubmed_sumpubmed_1")
159
+ #model_led_tldr, tokenizer_led_tldr = load_model_and_tokenizer("Blaise-g/led_large_sumpbumed_scitldr")
160
 
161
  name_to_path = load_example_filenames(_here / "examples")
162
  logging.info(f"Loaded {len(name_to_path)} examples")