Update app.py
Browse files
app.py
CHANGED
@@ -15,10 +15,10 @@ translation_models = {
|
|
15 |
'Spanish': "Helsinki-NLP/opus-mt-en-es"
|
16 |
}
|
17 |
|
18 |
-
#
|
19 |
summarization_models = {
|
20 |
'Scientific': "facebook/bart-large-cnn",
|
21 |
-
'Literature': "
|
22 |
}
|
23 |
|
24 |
# Initialize tokenizer
|
@@ -143,3 +143,4 @@ iface.launch()
|
|
143 |
|
144 |
|
145 |
|
|
|
|
15 |
'Spanish': "Helsinki-NLP/opus-mt-en-es"
|
16 |
}
|
17 |
|
18 |
+
# Summarization models
|
19 |
summarization_models = {
|
20 |
'Scientific': "facebook/bart-large-cnn",
|
21 |
+
'Literature': "google/pegasus-xsum"
|
22 |
}
|
23 |
|
24 |
# Initialize tokenizer
|
|
|
143 |
|
144 |
|
145 |
|
146 |
+
|