modified: app.py

app.py

@@ -245,7 +245,7 @@ class FluxEditor:
         return img, history_gallery
 
 
-    @spaces.GPU(duration=120)
+    @spaces.GPU(duration=200)
     @torch.inference_mode()
     def edit(self, init_image, source_prompt, target_prompt, editing_strategy, denoise_strategy, num_steps, guidance, attn_guidance_start_block, inject_step, init_image_2=None):
         
@@ -469,6 +469,7 @@ def on_change(init_image, changed: gr.EventData):
     img_path = list(changed.target.temp_files)
     return gr.Gallery(value=[(img_path[0], "")], label="History Image", interactive=True, columns=3), img_path[0]
 
+
 def create_demo(model_name: str, device: str = "cuda" if torch.cuda.is_available() else "cpu", offload: bool = False):
     editor = FluxEditor(args)
     is_schnell = model_name == "flux-schnell"
@@ -553,4 +554,4 @@ if __name__ == "__main__":
 
     demo = create_demo(args.name, args.device, args.offload)
     #demo.launch(server_name='0.0.0.0', share=args.share, server_port=args.port)
-    demo.launch(share=True)
+    demo.launch(debug=True)