codeteach commited on
Commit
fc9e651
1 Parent(s): f05ba0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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, device=0) # Ensure it runs on GPU for better performance
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
+