Update app.py

app.py

@@ -240,7 +240,7 @@ class main():
                 return image
         
         @spaces.GPU
-        def invert(image, mask, pcs=10000, epochs=400, weight_decay = 1e-10, lr=1e-1):
+        def invert(self, image, mask, pcs=10000, epochs=400, weight_decay = 1e-10, lr=1e-1):
             
             del unet
             del network
@@ -305,7 +305,7 @@ class main():
 
 
         @spaces.GPU
-        def run_inversion(dict, pcs, epochs, weight_decay,lr):
+        def run_inversion(self, dict, pcs, epochs, weight_decay,lr):
             init_image = dict["image"].convert("RGB").resize((512, 512))
             mask = dict["mask"].convert("RGB").resize((512, 512))
             network = invert([init_image], mask, pcs, epochs, weight_decay,lr)
@@ -323,10 +323,8 @@ class main():
     
     
         @spaces.GPU
-        def file_upload(file):
-            del unet
-            del network
-            
+        def file_upload(self, file):
+          
             proj = torch.load(file.name).to(device)
         
             #pad to 10000 Principal components to keep everything consistent
@@ -447,17 +445,17 @@ with gr.Blocks(css="style.css") as demo:
     
     
     
-        invert_button.click(fn=model.run_inversion(),
+        invert_button.click(fn=model.run_inversion,
                         inputs=[input_image, pcs, epochs, weight_decay,lr], 
                         outputs = [input_image, file_output])
             
                 
-        sample.click(fn=model.sample_then_run(), outputs=[input_image, file_output])
+        sample.click(fn=model.sample_then_run, outputs=[input_image, file_output])
             
         submit.click(
-                fn=model.edit_inference(), inputs=[prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4], outputs=[gallery]
+                fn=model.edit_inference, inputs=[prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4], outputs=[gallery]
             )
-        file_input.change(fn=model.file_upload(), inputs=file_input, outputs = gallery)
+        file_input.change(fn=model.file_upload, inputs=file_input, outputs = gallery)