Update app.py

app.py

@@ -389,7 +389,7 @@ def run_inversion(self, dict, pcs, epochs, weight_decay,lr):
     
     
 @spaces.GPU
-def file_upload(file):
+def file_upload(file, net):
     device="cuda"
     weights = torch.load(file.name).to(device)
         
@@ -511,7 +511,7 @@ with gr.Blocks(css="style.css") as demo:
         submit.click(
                 fn=edit_inference, inputs=[net, prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4], outputs=[net, gallery]
             )
-        file_input.change(fn=file_upload, inputs=[file_input, net], outputs = [net, gallery])
+        file_input.change(fn=file_upload, inputs=[file_input, net], outputs = [net, input_image])