Spaces:

DemocracyStudio
/

generate_nft_content

Build error

jcarbonnell commited on Jun 14, 2022

Commit

c415045

•

1 Parent(s): 2b360a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,8 +1,9 @@
 import streamlit as st
-from transformers import pipeline
 from summarizer import Summarizer
-pipeline = pipeline(task='text-generation', model="DemocracyStudio/generate_nft_content")
 summarize=Summarizer()
 def main():

 import streamlit as st
+from transformers import GPT2Tokenizer, GPT2LMHeadModel
 from summarizer import Summarizer
+model=GPT2LMHeadModel.from_pretrained("DemocracyStudio/generate_nft_content"))
+tokenizer=GPT2Tokenizer.from_pretrained("DemocracyStudio/generate_nft_content")
 summarize=Summarizer()
 def main():