to prevent more time spent building/uploading; you can upload the model to HF and then use that model id for it to be loaded
so in the app.py; you can put:model = GPT2LMHeadModel.from_pretrained('coffeeee/nsfw-stoy-blah')instead of the folder
model = GPT2LMHeadModel.from_pretrained('coffeeee/nsfw-stoy-blah')
· Sign up or log in to comment