Update app.py

app.py

@@ -1,6 +1,7 @@
 from stable_diffusion_tf.stable_diffusion import StableDiffusion as StableDiffusionPy
 import gradio as gr
 from tensorflow import keras
+from PIL import Image
 
 keras.mixed_precision.set_global_policy("float32")
 # load keras model
@@ -21,7 +22,9 @@ def generate_images(prompt: str, num_steps: int, unconditional_guidance_scale: i
         batch_size=1,
     )
 
-    return generated_img[0]
+    img = Image.fromarray(generated_img[0])
+
+    return img
 
 
 # pass function, input type for prompt, the output for multiple images