Spaces:

reshav1
/

test

Sleeping

test / app.py

Update app.py

139bee4 verified about 1 year ago

203 Bytes

	import transformers
	from transformers import AutoModelForCausalLM
	model_id = "gpt2"

	model = AutoModelForCausalLM.from_pretrained(model_id)

	text = "Hello"

	outputs = model.generate(text)
	print(outputs)