Update app.py
Browse files
app.py
CHANGED
@@ -10,8 +10,8 @@ translation_models = {
|
|
10 |
'Spanish': "Helsinki-NLP/opus-mt-en-es"
|
11 |
}
|
12 |
|
13 |
-
# Initialize summarization pipeline
|
14 |
-
summarizer = pipeline("summarization")
|
15 |
|
16 |
# Initialize translation pipeline
|
17 |
def get_translator(language):
|
@@ -71,3 +71,4 @@ iface = gr.Interface(
|
|
71 |
|
72 |
iface.launch()
|
73 |
|
|
|
|
10 |
'Spanish': "Helsinki-NLP/opus-mt-en-es"
|
11 |
}
|
12 |
|
13 |
+
# Initialize summarization pipeline with a specified model
|
14 |
+
summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
|
15 |
|
16 |
# Initialize translation pipeline
|
17 |
def get_translator(language):
|
|
|
71 |
|
72 |
iface.launch()
|
73 |
|
74 |
+
|