Update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,7 @@ translation_models = {
|
|
17 |
|
18 |
# Initialize summarization pipeline with a specified model
|
19 |
summarization_model = "facebook/bart-large-cnn"
|
20 |
-
summarizer = pipeline("summarization", model=summarization_model
|
21 |
tokenizer = AutoTokenizer.from_pretrained(summarization_model)
|
22 |
|
23 |
# Initialize translation pipeline
|
@@ -115,3 +115,4 @@ iface.launch()
|
|
115 |
|
116 |
|
117 |
|
|
|
|
17 |
|
18 |
# Initialize summarization pipeline with a specified model
|
19 |
summarization_model = "facebook/bart-large-cnn"
|
20 |
+
summarizer = pipeline("summarization", model=summarization_model) # Removed device parameter
|
21 |
tokenizer = AutoTokenizer.from_pretrained(summarization_model)
|
22 |
|
23 |
# Initialize translation pipeline
|
|
|
115 |
|
116 |
|
117 |
|
118 |
+
|