Update app.py

app.py

@@ -46,10 +46,10 @@ def generate_images(diffusion_steps, stochasticity, min_signal_rate, max_signal_
         )
         
     # denormalize
-    data_mean = tf.constant([[[[0.4704 0.3942 0.3033]]]])
-    data_variance = tf.constant([[[[0.08366 0.05590 0.07183]]]])
-    generated_images = data_mean + pred_images * data_variance ** 0.5
-    generated_images = tf.clip_by_value(pred_images, 0.0, 1.0)
+    data_mean = tf.constant([[[[0.4705 0.3943 0.3033]]]])
+    data_std_dev = tf.constant([[[[0.2892 0.2364 0.2680]]]])
+    generated_images = data_mean + pred_images * data_std_dev
+    generated_images = tf.clip_by_value(generated_images, 0.0, 1.0)
     
     # make grid
     generated_images = tf.image.resize(generated_images, (plot_image_size, plot_image_size), method="nearest")