Update app.py

app.py

@@ -144,7 +144,7 @@ def sample_then_run(net):
     cfg = 3.0
     steps = 25
     image = inference(net, prompt, negative_prompt, cfg, steps, seed)
-    return net,net,image
+    return net,net,image, None
 
 @torch.no_grad()
 @spaces.GPU()
@@ -401,7 +401,7 @@ def run_inversion(net, dict, pcs, epochs, weight_decay,lr):
     steps = 25
     image = inference( net, prompt, negative_prompt, cfg, steps, seed)
     
-    return net, net, image
+    return net, net, image, None
     
     
 @spaces.GPU
@@ -422,7 +422,7 @@ def file_upload(file, net):
     cfg = 3.0
     steps = 25
     image = inference(net, prompt, negative_prompt, cfg, steps, seed)
-    return net, image
+    return net, image, None
     
 help_text1 = """
             <b>Instructions</b>:
@@ -531,7 +531,7 @@ with gr.Blocks(css="style.css") as demo:
         gr.Markdown(help_text1)
         gr.Markdown(help_text2)
     
-    gr.Markdown("""<div style="text-align: justify;"> After sampling a new model or inverting, you can download the model below.""")
+    gr.Markdown("""<div style="text-align: justify;">❸ After sampling a new model or inverting, you can download the model below.""")
     
     with gr.Row():
         file_output = gr.File(label="Download Model", container=True, interactive=False)
@@ -541,15 +541,15 @@ with gr.Blocks(css="style.css") as demo:
     
         invert_button.click(fn=run_inversion,
                         inputs=[net, input_image, pcs, epochs, weight_decay,lr], 
-                        outputs = [net, file_output, input_image])
+                        outputs = [net, file_output, input_image, gallery])
             
                 
-        sample.click(fn=sample_then_run,inputs = [net], outputs=[net, file_output, input_image])
+        sample.click(fn=sample_then_run,inputs = [net], outputs=[net, file_output, input_image, gallery])
             
         submit.click(
                 fn=edit_inference, inputs=[net, prompt, negative_prompt, cfg, steps, seed, injection_step, a1, a2, a3, a4, input_image], outputs=[net, gallery]
             )
-        file_input.change(fn=file_upload, inputs=[file_input, net], outputs = [net, input_image])
+        file_input.change(fn=file_upload, inputs=[file_input, net], outputs = [net, input_image, gallery])