balaramas commited on
Commit
ab4c612
1 Parent(s): 63efd0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -3,6 +3,7 @@ import gradio as gr
3
  import subprocess
4
  subprocess.check_call(["pip", "install", "transformers"])
5
  subprocess.check_call(["pip", "install", "torch"])
 
6
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
7
  from transformers import pipeline
8
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
 
3
  import subprocess
4
  subprocess.check_call(["pip", "install", "transformers"])
5
  subprocess.check_call(["pip", "install", "torch"])
6
+ subprocess.check_call(["pip", "install", "sentencepiece"])
7
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
8
  from transformers import pipeline
9
  summarizer = pipeline("summarization", model="facebook/bart-large-cnn")