Update app.py
Browse files
app.py
CHANGED
@@ -341,7 +341,7 @@ def schleifer_model():
|
|
341 |
@st.experimental_singleton(suppress_st_warning=True)
|
342 |
def google_model():
|
343 |
model_name = 'google/pegasus-cnn_dailymail'
|
344 |
-
summarizer = pipeline('summarization',model=
|
345 |
device=0 if torch.cuda.is_available() else -1)
|
346 |
return summarizer
|
347 |
|
|
|
341 |
@st.experimental_singleton(suppress_st_warning=True)
|
342 |
def google_model():
|
343 |
model_name = 'google/pegasus-cnn_dailymail'
|
344 |
+
summarizer = pipeline('summarization',model=model_name, tokenizer=model_name,
|
345 |
device=0 if torch.cuda.is_available() else -1)
|
346 |
return summarizer
|
347 |
|