Update app.py

app.py

@@ -13,13 +13,13 @@ def generate_and_save_images(number_of_examples):
   seed = tf.random.normal([number_of_examples, 100])
   predictions = model(seed, training=False)
 
-  fig = plt.figure(figsize=(160, 640))
+  fig = plt.figure(figsize=(80, 80))
 
   for i in range(predictions.shape[0]):
-      plt.subplot(1, 4, i+1)
+      plt.subplot(2, 2, i+1)
       plt.imshow(predictions[i, :, :, :])
       plt.axis('off')
-  return plt
+  return fig