Spaces:
Running
Running
Update app.py
Browse filesmodel checkpoint update
app.py
CHANGED
@@ -8,8 +8,8 @@ import torch
|
|
8 |
def load_model():
|
9 |
"""Load the PEFT model and tokenizer once and cache them"""
|
10 |
base_model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
|
11 |
-
peft_model = PeftModel.from_pretrained(base_model, "Lakshan2003/finetuned-xsum")
|
12 |
-
tokenizer = AutoTokenizer.from_pretrained("Lakshan2003/finetuned-xsum")
|
13 |
return peft_model, tokenizer
|
14 |
|
15 |
|
|
|
8 |
def load_model():
|
9 |
"""Load the PEFT model and tokenizer once and cache them"""
|
10 |
base_model = AutoModelForSeq2SeqLM.from_pretrained("t5-small")
|
11 |
+
peft_model = PeftModel.from_pretrained(base_model, "Lakshan2003/finetuned-t5-xsum")
|
12 |
+
tokenizer = AutoTokenizer.from_pretrained("Lakshan2003/finetuned-t5-xsum")
|
13 |
return peft_model, tokenizer
|
14 |
|
15 |
|