Update app.py

app.py

@@ -13,11 +13,8 @@ with gr.Blocks() as demo:
     """
     * Train/Eval will setup, train, and evaluate the base model
     """)
-    def modelTraining():
-        
-
-        print("TensorFlow version:", tf.__version__)
 
+    def modelTraining():
         mnist = tf.keras.datasets.mnist
 
         (x_train, y_train), (x_test, y_test) = mnist.load_data()
@@ -31,6 +28,8 @@ with gr.Blocks() as demo:
         ])
 
         predictions = model(x_train[:1]).numpy()
+        
+        print("TensorFlow version:", tf.__version__)
         print(predictions)
         
         tf.nn.softmax(predictions).numpy()