Update app.py

app.py

@@ -87,6 +87,7 @@ def main():
     @torch.no_grad()
     @spaces.GPU
     def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
+        nonlocal network
         generator = torch.Generator(device=device).manual_seed(seed)
         latents = torch.randn(
             (1, unet.in_channels, 512 // 8, 512 // 8),
@@ -131,6 +132,7 @@ def main():
     @torch.no_grad()
     @spaces.GPU
     def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4): 
+        nonlocal network
         original_weights = network.proj.clone()
         
         #pad to same number of PCs