Update app.py

app.py

@@ -137,7 +137,7 @@ def inference( prompt, negative_prompt, guidance_scale, ddim_steps, seed):
 @torch.no_grad()
 @spaces.GPU
 def edit_inference(prompt, negative_prompt, guidance_scale, ddim_steps, seed, start_noise, a1, a2, a3, a4):
-    print(network.value)
+    
    
     original_weights = network.value.proj.clone()
     
@@ -221,7 +221,7 @@ def sample_then_run():
     steps = 25
     image = inference( prompt, negative_prompt, cfg, steps, seed)
     torch.save(network.value.proj, "model.pt" )
-    return image, "model.pt", network.value
+    return image, "model.pt", network.value.cpu()
 
 
 class CustomImageDataset(Dataset):