Update app.py

app.py

@@ -53,7 +53,7 @@ def generate(image_editor, prompt, neg_prompt, versions):
     final_dict = {'background': final_image, 'layers': None, 'composite': final_image}
 
     if(versions==None): 
-        final_gallery = [final_image]
+        final_gallery = [image_editor['background'] ,final_image]
     else: 
         final_gallery = versions
         final_gallery.append(final_image)