Update app.py

app.py

@@ -2,6 +2,7 @@ import gradio as gr
 import torch
 from diffusers import StableDiffusionPipeline
 import boto3
+from io import BytesIO
 import os
 
 AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")