Update app.py

app.py

@@ -9,6 +9,7 @@ from huggingface_hub import hf_hub_download
 import requests
 import gradio as gr
 import numpy as np
+from PIL import Image
 
 class Upsample(nn.Module):
     def __init__(self, in_channels, out_channels, kernel_size=4, stride=2, padding=1, dropout=True):