codeteach commited on
Commit
128d6a6
1 Parent(s): 002b4f1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -17,12 +17,12 @@ translation_models = {
17
 
18
  # Initialize summarization pipelines with specified models
19
  summarization_models = {
20
- 'Scientific': "sshleifer/distilbart-cnn-12-6",
21
  'Literature': "t5-small"
22
  }
23
 
24
  # Initialize tokenizer
25
- tokenizer = AutoTokenizer.from_pretrained("t5-small") # Using t5-small tokenizer for all models
26
 
27
  # Initialize summarization pipelines
28
  summarizers = {model: pipeline("summarization", model=model) for model in summarization_models.values()}
@@ -141,3 +141,4 @@ iface.launch()
141
 
142
 
143
 
 
 
17
 
18
  # Initialize summarization pipelines with specified models
19
  summarization_models = {
20
+ 'Scientific': "facebook/bart-large-cnn",
21
  'Literature': "t5-small"
22
  }
23
 
24
  # Initialize tokenizer
25
+ tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn") # Using BART tokenizer for the 'Scientific' model
26
 
27
  # Initialize summarization pipelines
28
  summarizers = {model: pipeline("summarization", model=model) for model in summarization_models.values()}
 
141
 
142
 
143
 
144
+