# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("eugenesiow/edsr")