Update app.py

app.py

@@ -96,7 +96,7 @@ def sample_then_run(network, unet):
     image = inference( network, unet, prompt, negative_prompt, cfg, steps, seed)
     torch.save(network.proj.detach().cpu(), "model.pt" )
     print("done saving")
-    return image, "model.pt", unet, network.cpu()
+    return image, "model.pt", unet.cpu(), network.cpu()