Spaces:

Erfan11
/

Textwizai

Sleeping

Textwizai / load_model.py

Create load_model.py

0e7cada verified about 2 months ago

210 Bytes

	import torch
	from transformers import AutoModelForSequenceClassification

	def load_model(model_path):
	model = AutoModelForSequenceClassification.from_pretrained(model_path)
	model.eval()
	return model