Update app.py

app.py

@@ -166,7 +166,7 @@ def preview_image_and_mask(image, width, height, overlap_percentage, resize_opti
     
     return preview
 
-@spaces.GPU(duration=20)
+@spaces.GPU(duration=24)
 def infer(image, width, height, overlap_percentage, num_inference_steps, resize_option, custom_resize_percentage, prompt_input, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom):
     background, mask = prepare_image_and_mask(image, width, height, overlap_percentage, resize_option, custom_resize_percentage, alignment, overlap_left, overlap_right, overlap_top, overlap_bottom)
     
@@ -245,24 +245,10 @@ def update_history(new_image, history):
 
 css = """
 .gradio-container {
-    max-width: 1200px !important;
-    margin: auto !important;
-}
-/* Add some spacing */
-.gr-form > * {
-    margin-bottom: 10px;
-}
-.gr-button {
-    min-width: 100px; /* Ensure buttons have a minimum width */
-}
-/* Style the history gallery */
-#history_gallery .gr-gallery {
-    min-height: 150px; /* Give gallery some minimum height */
-}
-#history_gallery .gr-gallery > .gr-image {
-    border: 1px solid #ddd; /* Add subtle border to gallery images */
-    border-radius: 4px;
+    width: 1200px !important;
 }
+h1 { text-align: center; }
+footer { visibility: hidden; }
 """
 
 title = """<h1 align="center">Diffusers Image Outpaint Lightning</h1>
@@ -362,8 +348,10 @@ with gr.Blocks(css=css) as demo:
 
             with gr.Column():
                 result = gr.Image(label="Generated Image", format="png")
+                use_as_input_button = gr.Button("Use as Input Image", visible=False, interactive=False)
 
                 history_gallery = gr.Gallery(label="History", columns=6, object_fit="contain", interactive=False)
+                preview_image = gr.Image(label="Preview")
 
     def use_output_as_input(output_image):
         """Sets the generated output as the new input image."""