Komal-patra commited on
Commit
23a30f1
·
verified ·
1 Parent(s): c28a9ce

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -11,9 +11,9 @@ max_length=512
11
 
12
  MODEL_NAME = "google/flan-t5-base"
13
  tokenizer = T5Tokenizer.from_pretrained(MODEL_NAME, token=token)
14
- config = PeftConfig.from_pretrained("Orcawise/results")
15
  base_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
16
- model = PeftModel.from_pretrained(base_model, "Orcawise/results")
17
 
18
  #gr.Interface.from_pipeline(pipe).launch()
19
 
 
11
 
12
  MODEL_NAME = "google/flan-t5-base"
13
  tokenizer = T5Tokenizer.from_pretrained(MODEL_NAME, token=token)
14
+ config = PeftConfig.from_pretrained("Komal-patra/results")
15
  base_model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
16
+ model = PeftModel.from_pretrained(base_model, "Komal-patra/results")
17
 
18
  #gr.Interface.from_pipeline(pipe).launch()
19