Update app.py

app.py

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