Dhanush S Gowda
commited on
Update app.py
Browse files
app.py
CHANGED
@@ -12,7 +12,7 @@ def load_model(model_name):
|
|
12 |
elif model_name == 'Pegasus':
|
13 |
return pipeline("summarization", model="google/pegasus-cnn_dailymail")
|
14 |
|
15 |
-
# Function to load all models concurrently
|
16 |
@st.cache_resource
|
17 |
def load_all_models():
|
18 |
model_names = ['BART', 'T5', 'Pegasus']
|
|
|
12 |
elif model_name == 'Pegasus':
|
13 |
return pipeline("summarization", model="google/pegasus-cnn_dailymail")
|
14 |
|
15 |
+
# Function to load all models concurrently and cache them
|
16 |
@st.cache_resource
|
17 |
def load_all_models():
|
18 |
model_names = ['BART', 'T5', 'Pegasus']
|