Initial Code Commit
This commit is contained in:
6
AI Training/hf.py
Executable file
6
AI Training/hf.py
Executable file
@@ -0,0 +1,6 @@
|
||||
# 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")
|
||||
|
||||
Reference in New Issue
Block a user