Qwelian D Tanner qweliant

🧟

A zombie, philosophically speaking

My interest involves systems and collective decision-making. I wanna be a better baker.

qweliant / nlp.py

Last active September 1, 2020 17:19

NLP{ class for modeling

	from transformers import (
	pipeline,
	GPT2LMHeadModel,
	GPT2Tokenizer
	)

	class NLP:
	def __init__(self):
	self.gen_model = GPT2LMHeadModel.from_pretrained('gpt2')
	self.gen_tokenizer = GPT2Tokenizer.from_pretrained('gpt2')