Update app.py

app.py

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