checkiejan commited on
Commit
992173c
·
1 Parent(s): 2a3d663

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -8,11 +8,12 @@ import datasets
8
  # Load your fine-tuned model and tokenizer
9
  model_name = "google/flan-t5-large"
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
11
- model = T5ForConditionalGeneration.from_pretrained("google/flan-t5-large")
12
- model.load_adapter("legacy107/adapter-flan-t5-large-bottleneck-adapter-cpgQA", source="hf")
13
- model.set_active_adapters("question_answering")
 
14
 
15
- peft_name = "legacy107/flan-t5-large-ia3-bioasq-paraphrase"
16
  peft_config = PeftConfig.from_pretrained(peft_name)
17
  paraphrase_model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
18
  paraphrase_model = PeftModel.from_pretrained(paraphrase_model, peft_name)
 
8
  # Load your fine-tuned model and tokenizer
9
  model_name = "google/flan-t5-large"
10
  tokenizer = AutoTokenizer.from_pretrained(model_name)
11
+ peft_name = "checkiejan/flan-t5-prefix-25-9-2"
12
+ peft_config = PeftConfig.from_pretrained(peft_name)
13
+ model = AutoModelForSeq2SeqLM.from_pretrained(peft_config.base_model_name_or_path)
14
+ model = PeftModel.from_pretrained(model, peft_name)
15
 
16
+ peft_name = "checkiejan/prefix-paraphase-50-19-auto"
17
  peft_config = PeftConfig.from_pretrained(peft_name)
18
  paraphrase_model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
19
  paraphrase_model = PeftModel.from_pretrained(paraphrase_model, peft_name)