Update app.py

app.py

@@ -83,6 +83,7 @@ def main():
         proj.to(device)
         unet, _, _, _, _ = load_models(device)
         network = sample_weights(unet, proj, mean, std, v[:, :1000], device, factor = 1.00)
+        return network
  
     @torch.no_grad()
     @spaces.GPU
@@ -206,7 +207,8 @@ def main():
 
     @spaces.GPU
     def sample_then_run():
-        sample_model()    
+        nonlocal network 
+        network = sample_model()    
         prompt = "sks person"
         negative_prompt = "low quality, blurry, unfinished, nudity, weapon"
         seed = 5