Update app.py

app.py

@@ -32,13 +32,14 @@ weight_dimensions = torch.load(f"{models_path}/weight_dimensions.pt")
 
 unet, vae, text_encoder, tokenizer, noise_scheduler = load_models(device)
 network = sample_weights(unet, proj, mean, std, v[:, :1000], device, factor = 1.00)
-#global network
+global network
 
 def sample_model():
    global unet
    del unet
    global network
    unet, _, _, _, _ = load_models(device)
+    
 @torch.no_grad()
 def inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed):
     global device