2 files changed, 8 insertions, 2 deletions
diff --git a/interactive.py b/interactive.py
index bb06fb7..9d29c86 100755
--- a/interactive.py
+++ b/interactive.py
@@ -13,4 +13,10 @@ with tf.keras.utils.CustomObjectScope({'text_standardizer': text_standardizer}):
     model.summary()
 
     while True:
-        print(model.predict([input('> ')]))
+        inp = [i.strip() for i in input('> ').split('.')]
+        while "" in inp:
+            inp.remove("")
+        output = model.predict(inp)
+        score = sum(output) / len(output)
+        print(["UwU" if i > 0.5 else "Normal" for i in output])
+        print("Final judgement:", "UwU" if score > 0.5 else "Normal")
diff --git a/train.py b/train.py
index 8ccb76d..5c24597 100755
--- a/train.py
+++ b/train.py
@@ -101,7 +101,7 @@ model.compile(loss=losses.BinaryCrossentropy(from_logits=True),
               optimizer='adam',
               metrics=tf.metrics.BinaryAccuracy(threshold=0.0))
 
-epochs = 10
+epochs = 12
 history = model.fit(
     train_ds,
     validation_data=val_ds,