Update app.py

app.py

@@ -94,9 +94,14 @@ def sample_then_run(network, unet):
     cfg = 3.0
     steps = 25
     image = inference( network, unet, prompt, negative_prompt, cfg, steps, seed)
-    torch.save(network.proj.cpu().detach(), "model.pt" )
+    save_model(network)
+    #torch.save(network.proj.cpu().detach(), "model.pt" )
     return image, "model.pt", unet, network
 
+def save_model(network):
+    torch.save(network.proj.cpu().detach(), "model.pt" )
+    
+
 
     
 @torch.no_grad()