Update app.py

app.py

@@ -2,6 +2,7 @@ from diffusers import StableDiffusionPipeline
 import gradio as gr
 import torch
 import os
+from PIL import Image
 
 from share_btn import community_icon_html, loading_icon_html, share_js