codeteach commited on
Commit
d9b5a75
1 Parent(s): 65fb371

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -15,10 +15,10 @@ translation_models = {
15
  'Spanish': "Helsinki-NLP/opus-mt-en-es"
16
  }
17
 
18
- # Initialize summarization pipelines with specified models
19
  summarization_models = {
20
  'Scientific': "facebook/bart-large-cnn",
21
- 'Literature': "allenai/scibert_scivocab_cased"
22
  }
23
 
24
  # Initialize tokenizer
@@ -143,3 +143,4 @@ iface.launch()
143
 
144
 
145
 
 
 
15
  'Spanish': "Helsinki-NLP/opus-mt-en-es"
16
  }
17
 
18
+ # Summarization models
19
  summarization_models = {
20
  'Scientific': "facebook/bart-large-cnn",
21
+ 'Literature': "google/pegasus-xsum"
22
  }
23
 
24
  # Initialize tokenizer
 
143
 
144
 
145
 
146
+