Spaces:
Sleeping
Sleeping
app.py
CHANGED
@@ -16,10 +16,9 @@ model_choices = {
|
|
16 |
"T5 Small (t5-small)": "t5-small",
|
17 |
"T5 Base (t5-base)": "t5-base",
|
18 |
"Flan-T5 Base (google/flan-t5-base)": "google/flan-t5-base",
|
19 |
-
"DistilBART XSum (
|
20 |
}
|
21 |
|
22 |
-
|
23 |
model_cache = {}
|
24 |
|
25 |
def clean_text(input_text):
|
|
|
16 |
"T5 Small (t5-small)": "t5-small",
|
17 |
"T5 Base (t5-base)": "t5-base",
|
18 |
"Flan-T5 Base (google/flan-t5-base)": "google/flan-t5-base",
|
19 |
+
"DistilBART XSum (sshleifer/distilbart-xsum-12-6)": "sshleifer/distilbart-xsum-12-6"
|
20 |
}
|
21 |
|
|
|
22 |
model_cache = {}
|
23 |
|
24 |
def clean_text(input_text):
|