File size: 345 Bytes
6af334f |
1 2 3 4 5 6 7 8 9 10 11 |
from transformers import AutoModelForImageClassification, AutoImageProcessor
model_name = "Asmaa111/diabetic-eye"
save_path = "./diabetic_model"
model = AutoModelForImageClassification.from_pretrained(model_name)
processor = AutoImageProcessor.from_pretrained(model_name)
model.save_pretrained(save_path)
processor.save_pretrained(save_path)
|