Update app.py

app.py

@@ -55,6 +55,9 @@ def predict(prompt, upload_images, ip_adapter_scale=0.5, negative_prompt="", see
 def swap_to_gallery(images):
     return gr.update(value=images, visible=True), gr.update(visible=True), gr.update(visible=False)
 
+def remove_back_to_files():
+    return gr.update(visible=False), gr.update(visible=False), gr.update(visible=True)
+
 examples = [
     ["high quality", "example1.png", 1.0, "", 1000, False, False, 1152, 896],
     ["capybara", "example2.png", 0.7, "", 1000, False, False, 1152, 896],