Update app.py

app.py

@@ -20,9 +20,10 @@ print("downloading models - 4/4")
 hf_hub_download("XLabs-AI/flux-controlnet-canny", "controlnet.safetensors")
 print("downloaded!")
 
-@spaces.GPU(duration=240)
+from src.flux.xflux_pipeline import XFluxPipeline
+
+@spaces.GPU(duration=200)
 def process_image(lora_path, lora_name, number_of_images, image, prompt, steps, use_lora, use_controlnet, use_depth, use_hed, use_ip, lora_weight, negative_image, neg_prompt, true_gs, guidance, cfg):
-    from src.flux.xflux_pipeline import XFluxPipeline
     def run_xflux_pipeline(
         prompt, image, repo_id, name, device,
         model_type, width, height, timestep_to_start_cfg, num_steps, true_gs, guidance,
@@ -139,58 +140,7 @@ def process_image(lora_path, lora_name, number_of_images, image, prompt, steps,
       )
 
 
-custom_css = """
-body {
-  overflow-y: hidden;
-}
-.gradio-container {
-  margin: unset !important;
-  padding: unset !important;
-  max-width: unset !important;
-}
-.gradio-app {
-  min-height: unset !important;
-}
-
-.app {
-  height: 100%;
-}
-
-.app-container {
-  height: 100vh;
-  background: rgb(24, 24, 27);
-  gap: unset;
-}
-
-.sidebar {
-  overflow-y: auto;
-  height: 100%;
-  max-width: 500px;
-  padding: 1rem;
-  background: rgb(31, 31, 35);
-  border-right: 1px solid rgb(41, 41, 41);
-}
-
-.grid-wrap {
-  max-height: unset;
-  overflow-y: unset;
-}
-
-.thumbnail-item img {
-  object-fit: contain;
-}
-
-.side_items {
-  height: auto;
-}
-
-.form {
-  flex-wrap: unset;
-  height: auto;
-}
-"""
-
-with gr.Blocks(css=custom_css) as demo:
+with gr.Blocks() as demo:
     with gr.Row(elem_classes="app-container"):
         with gr.Column(scale=1, min_width=500, elem_classes="sidebar"):
             with gr.Column(elem_classes="side_items"):
@@ -208,7 +158,6 @@ with gr.Blocks(css=custom_css) as demo:
                 lora_name = gr.Textbox(label="Lora Name", value="realism_lora.safetensors")
                 lora_weight = gr.Slider(step=0.1, minimum=0, maximum=1, value=0.7, label="Lora Weight")
                 
-                number_of_images = gr.Slider(step=1, minimum=0, maximum=4, value=2, label="Number of Images")
                 true_gs = gr.Slider(step=0.1, minimum=0, maximum=10, value=3.5, label="TrueGs")
                 guidance = gr.Slider(minimum=1, maximum=10, value=4, label="Guidance")
                 cfg = gr.Slider(minimum=1, maximum=10, value=1, label="CFG")
@@ -219,7 +168,7 @@ with gr.Blocks(css=custom_css) as demo:
         with gr.Column(scale=2, elem_classes="app"):
             output = gr.Gallery(label="Galery output", elem_classes="galery")
 
-    submit_btn.click(process_image, inputs=[lora_path, lora_name, number_of_images, input_image, prompt, steps, use_lora, controlnet, use_depth, use_hed, use_ip, lora_weight, negative_image, neg_prompt, true_gs, guidance, cfg], outputs=output)
+    submit_btn.click(process_image, inputs=[lora_path, lora_name, 1, input_image, prompt, steps, use_lora, controlnet, use_depth, use_hed, use_ip, lora_weight, negative_image, neg_prompt, true_gs, guidance, cfg], outputs=output)
 
 if __name__ == '__main__':
     demo.launch(share=True, debug=True)