Update app.py
Browse files
app.py
CHANGED
@@ -319,7 +319,7 @@ def schleifer_model():
|
|
319 |
|
320 |
@st.experimental_singleton(suppress_st_warning=True)
|
321 |
def google_model():
|
322 |
-
model_name = 'google/pegasus-
|
323 |
summarizer = pipeline('summarization',model=model_name, tokenizer=model_name,
|
324 |
device=0 if torch.cuda.is_available() else -1)
|
325 |
return summarizer
|
|
|
319 |
|
320 |
@st.experimental_singleton(suppress_st_warning=True)
|
321 |
def google_model():
|
322 |
+
model_name = 'google/pegasus-large'
|
323 |
summarizer = pipeline('summarization',model=model_name, tokenizer=model_name,
|
324 |
device=0 if torch.cuda.is_available() else -1)
|
325 |
return summarizer
|