.

options/Banner_Model/Image2Image.py

@@ -39,7 +39,7 @@ def resize_image_dimensions(
     return new_width, new_height
 
 
-@spaces.GPU
+@spaces.GPU(duration=100)
 def I2I(
     input_image_editor: dict,
     input_text: str,
@@ -65,7 +65,7 @@ def I2I(
         return None, None
     
     pipe = FluxInpaintPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-dev", torch_dtype=torch.bfloat16).to(DEVICE)
+    "black-forest-labs/FLUX.1-schnell", torch_dtype=torch.bfloat16).to(DEVICE)
 
     width, height = resize_image_dimensions(original_resolution_wh=image.size)
     resized_image = image.resize((width, height), Image.LANCZOS)