import tensorflow as tf
from model import create_model
import numpy as np

def load_data():
    return np.random.rand(1000,64,64,3).astype(np.float32), np.random.randint(10, size=1000)

def main():
    train_images, train_labels=load_data()
    model = create_model()
    model.fit(train_images, train_labels, epochs=5)
    model.save('my_model.keras')

if __name__ == "__main__":
    main()