Spaces:

XaKnee
/

LLMTWO

Runtime error

LLMTWO / model_loader.py

Update model_loader.py

4b857ac verified over 1 year ago

128 Bytes

from transformers import pipeline

def load_model():
    pipe = pipeline("text-generation", model="distilgpt2")
    return pipe

	from transformers import pipeline

	def load_model():
	pipe = pipeline("text-generation", model="distilgpt2")
	return pipe