Spaces:

riyadifirman
/

klasifikasiburung_1

Sleeping

riyadifirman commited on Oct 8

Commit

2a01cbf

•

1 Parent(s): 20115ca

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,6 +3,7 @@ import torch
 from transformers import AutoImageProcessor, AutoModelForImageClassification
 from torchvision.transforms import Compose, Resize, ToTensor, Normalize, RandomHorizontalFlip, RandomRotation
 from PIL import Image
 import traceback
 # Load dataset to get labels

 from transformers import AutoImageProcessor, AutoModelForImageClassification
 from torchvision.transforms import Compose, Resize, ToTensor, Normalize, RandomHorizontalFlip, RandomRotation
 from PIL import Image
+from datasets import load_dataset
 import traceback
 # Load dataset to get labels