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(4096, activation = "relu"),
    l.Dense(4096, activation = "relu"),
    l.Dense(4096, activation = "relu"),
    l.Dense(10, activation = "softmax")
])

#f.train(m, "3")

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