Spaces:
Runtime error
Runtime error
Abinaya Mahendiran
commited on
Commit
·
ebee998
1
Parent(s):
695e732
Updated app
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ st.set_page_config(layout="wide")
|
|
20 |
def load_model():
|
21 |
tokenizer = AutoTokenizer.from_pretrained(cfg["model_name_or_path"])
|
22 |
model = GPT2LMHeadModel.from_pretrained(cfg["model_name_or_path"])
|
23 |
-
generator = pipeline("
|
24 |
return generator, tokenizer
|
25 |
|
26 |
with st.spinner('Loading model...'):
|
|
|
20 |
def load_model():
|
21 |
tokenizer = AutoTokenizer.from_pretrained(cfg["model_name_or_path"])
|
22 |
model = GPT2LMHeadModel.from_pretrained(cfg["model_name_or_path"])
|
23 |
+
generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
|
24 |
return generator, tokenizer
|
25 |
|
26 |
with st.spinner('Loading model...'):
|