codeteach commited on
Commit
fb658f7
1 Parent(s): b588b2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
+