Spaces:

lityops
/

Abstractive-Style-Summarizer

Running

lityops commited on Jan 4

Commit

9fb94a5

verified ·

1 Parent(s): 89793b6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from peft import PeftModel
 ADAPTER_HUB = "lityops/Abstractive-Style-Summarizer"
 BASE_MODEL_NAME = "google/flan-t5-base"
-base_model = AutoModelForSeq2SeqLM.from_pretrained(BASE_MODEL)
 model = PeftModel.from_pretrained(base_model, ADAPTER_HUB)
 tokenizer = AutoTokenizer.from_pretrained(ADAPTER_HUB)

 ADAPTER_HUB = "lityops/Abstractive-Style-Summarizer"
 BASE_MODEL_NAME = "google/flan-t5-base"
+base_model = AutoModelForSeq2SeqLM.from_pretrained(BASE_MODEL_NAME)
 model = PeftModel.from_pretrained(base_model, ADAPTER_HUB)
 tokenizer = AutoTokenizer.from_pretrained(ADAPTER_HUB)