DiabeticRetionPathyDetection / download_model.py
AsmaaElnagger's picture
init
6af334f
raw
history blame contribute delete
345 Bytes
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)