Update app.py
Browse files
app.py
CHANGED
@@ -3,8 +3,8 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
|
|
3 |
|
4 |
# Load tokenizer and model
|
5 |
model_identifier = "songhieng/khmer-mt5-summarization"
|
6 |
-
tokenizer = AutoTokenizer.from_pretrained(model_identifier)
|
7 |
-
model = AutoModelForSeq2SeqLM.from_pretrained(model_identifier)
|
8 |
|
9 |
# Set page configuration
|
10 |
st.set_page_config(page_title="Khmer Text Summarization", layout="wide")
|
|
|
3 |
|
4 |
# Load tokenizer and model
|
5 |
model_identifier = "songhieng/khmer-mt5-summarization"
|
6 |
+
tokenizer = AutoTokenizer.from_pretrained(model_identifier, use_fast=False)
|
7 |
+
model = AutoModelForSeq2SeqLM.from_pretrained(model_identifier, use_fast=False)
|
8 |
|
9 |
# Set page configuration
|
10 |
st.set_page_config(page_title="Khmer Text Summarization", layout="wide")
|