Update app.py

app.py

@@ -303,10 +303,9 @@ def sample_then_run():
     cfg = 3.0
     steps = 25
     image = inference( prompt, negative_prompt, cfg, steps, seed)
-    print("doneeee111111")
     torch.save(network.value.proj, "model.pt" )
-    print("doneeee22222222")
-    return image, "model.pt", network.value.cpu()
+    
+    return image, "model.pt"   #, network.value.cpu()
 
 
 class CustomImageDataset(Dataset):
@@ -540,7 +539,7 @@ with gr.Blocks(css="style.css") as demo:
                     outputs = [input_image, file_output])
         
             
-        sample.click(fn=sample_then_run, outputs=[input_image, file_output, network])
+        sample.click(fn=sample_then_run, outputs=[input_image, file_output])
         
         submit.click(
             fn=edit_inference, inputs=[prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4], outputs=[gallery]