Update app.py

app.py

@@ -305,9 +305,9 @@ def sample_then_run():
     image = inference( prompt, negative_prompt, cfg, steps, seed)
     torch.save(network.value.proj.detach(), "model.pt" )
     # net = torch.load("model.pt").cpu()
+    network.value.proj.detach().cpu()
     
-    
-    return image, "model.pt", network.value.cpu().detach() #net   #, network.value.cpu()
+    return image, "model.pt", network.value #net   #, network.value.cpu()
 
 
 class CustomImageDataset(Dataset):