Update

train_model.py

@@ -9,7 +9,9 @@ from tensorflow.keras.optimizers import Adam
 from tqdm import tqdm
 
 # Load dataset dari Hugging Face
-dataset = load_dataset("tanganke/stanford_cars")
+from datasets import load_dataset
+
+ds = load_dataset("tanganke/stanford_cars")
 NUM_CLASSES = 196
 IMG_SIZE = 224
 BATCH_SIZE = 32
@@ -22,8 +24,8 @@ def preprocess(example):
     return image, example["label"]
 
 # Apply preprocessing
-X_train, y_train = zip(*[preprocess(x) for x in tqdm(dataset["train"])])
-X_test, y_test = zip(*[preprocess(x) for x in tqdm(dataset["test"])])
+X_train, y_train = zip(*[preprocess(x) for x in tqdm(ds["train"])])
+X_test, y_test = zip(*[preprocess(x) for x in tqdm(ds["test"])])
 
 X_train = np.array(X_train)
 X_test = np.array(X_test)