Upload app.py

app.py

@@ -11,6 +11,7 @@ import torch
 from inference import inference
 from diffusers import StableDiffusionInstructPix2PixPipeline, UNet2DModel, AutoencoderKL, DDPMScheduler 
 from transformers import CLIPTokenizer, CLIPTextModel, CLIPImageProcessor
+import torch.nn as nn
 
 class UNetNoCondWrapper(nn.Module):
     def __init__(self, base_unet: UNet2DModel):