import tensorflow as tf
import tensorflow.keras.layers as l
from sys import argv

import f

m = tf.keras.models.Sequential([
    l.Input(shape = (784,)),
    l.Dense(128, activation = "relu"),
    l.Dense(64, activation = "relu"),
    l.Dense(10, activation = "softmax")
])

model_label = "1"

#f.train(m, model_label)
f.model_quality(m, model_label)

if len(argv) == 2:
    f.classify(m, model_label, argv[1])
else:
    f.classify_live(m, model_label)