Spaces:
Running
Running
import torch | |
from transformers import AutoModelForSequenceClassification | |
def load_model(model_path): | |
model = AutoModelForSequenceClassification.from_pretrained(model_path) | |
model.eval() | |
return model |