Lakshan2003 commited on
Commit
021abb6
·
verified ·
1 Parent(s): 8fec1e2

Update app.py

Browse files

model checkpoint update

Files changed (1) hide show
  1. app.py +2 -2
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