7 lines
314 B
Python
Executable File
7 lines
314 B
Python
Executable File
# Load model directly
|
|
from transformers import AutoImageProcessor, AutoModelForImageClassification
|
|
|
|
processor = AutoImageProcessor.from_pretrained("edwinpalegre/ee8225-group4-vit-trashnet-enhanced")
|
|
model = AutoModelForImageClassification.from_pretrained("edwinpalegre/ee8225-group4-vit-trashnet-enhanced")
|
|
|