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
|