Update app.py
Browse files
app.py
CHANGED
@@ -72,18 +72,18 @@ def proc_submission(
|
|
72 |
msg = None
|
73 |
|
74 |
if (model_type == "LED" and summary_type == "Detailed"):
|
75 |
-
|
76 |
elif (model_type == "LED" and summary_type == "TLDR"):
|
77 |
-
|
78 |
elif (model_type == "LongT5" and summary_type == "Detailed"):
|
79 |
-
|
80 |
else:
|
81 |
-
|
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 =
|
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")
|