reydeuss commited on
Commit
671de3c
·
verified ·
1 Parent(s): 4d52ef0

updated app.py to use the peft adapter from huggingface's model repo

Browse files
Files changed (1) hide show
  1. app.py +4 -6
app.py CHANGED
@@ -20,12 +20,10 @@ def load_model_once():
20
  device_map="auto"
21
  )
22
 
23
- # Load your trained PEFT adapters
24
- # Replace with your actual PEFT adapter path
25
- # model = PeftModel.from_pretrained(
26
- # base_model,
27
- # "./path-to-your-adapter" # Upload your wandb artifact files here
28
- # )
29
  return model, tokenizer
30
 
31
  def summarize_text(text):
 
20
  device_map="auto"
21
  )
22
 
23
+ model = PeftModel.from_pretrained(
24
+ base_model,
25
+ "reydeuss/trustify-t5-adapter",
26
+ )
 
 
27
  return model, tokenizer
28
 
29
  def summarize_text(text):