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 |
+
|