Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -47,15 +47,15 @@ def load_models():
|
|
47 |
# Use pipeline if available, now using the updated model
|
48 |
models['summarizer'] = pipeline(
|
49 |
"summarization",
|
50 |
-
model="
|
51 |
max_length=100,
|
52 |
truncation=True
|
53 |
)
|
54 |
else:
|
55 |
# Fall back to basic model loading using the updated summarization model
|
56 |
try:
|
57 |
-
models['summarizer_model'] = AutoModelForSeq2SeqLM.from_pretrained("
|
58 |
-
models['summarizer_tokenizer'] = AutoTokenizer.from_pretrained("
|
59 |
except Exception as e:
|
60 |
st.error(f"Error loading summarization model: {e}")
|
61 |
models['summarizer_model'] = None
|
|
|
47 |
# Use pipeline if available, now using the updated model
|
48 |
models['summarizer'] = pipeline(
|
49 |
"summarization",
|
50 |
+
model="Falconsai/text_summarization",
|
51 |
max_length=100,
|
52 |
truncation=True
|
53 |
)
|
54 |
else:
|
55 |
# Fall back to basic model loading using the updated summarization model
|
56 |
try:
|
57 |
+
models['summarizer_model'] = AutoModelForSeq2SeqLM.from_pretrained("Falconsai/text_summarization")
|
58 |
+
models['summarizer_tokenizer'] = AutoTokenizer.from_pretrained("Falconsai/text_summarization")
|
59 |
except Exception as e:
|
60 |
st.error(f"Error loading summarization model: {e}")
|
61 |
models['summarizer_model'] = None
|