feat: try just flux

app.py

@@ -14,7 +14,7 @@ from utils.sd3 import sd3_editing
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 
-pipe_sd3 = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
+# pipe_sd3 = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3-medium-diffusers", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
 pipe_flux = FluxPipeline.from_pretrained("black-forest-labs/FLUX.1-dev", torch_dtype=torch.float16, token=os.getenv('HF_ACCESS_TOK'))
 
 
@@ -74,7 +74,7 @@ def get_examples():
     ]
     return case
 
-@spaces.GPU(duration=200)
+@spaces.GPU(duration=240)
 def FlowOpt_run(
     image_src_val: str, model_type_val: str, T_steps_val: int,
     n_max_val: int, eta_val: float, flowopt_iterations_val: int,
@@ -88,8 +88,8 @@ def FlowOpt_run(
 
     if model_type_val == 'FLUX':
         pipe = pipe_flux.to(device)
-    elif model_type_val == 'SD3':
-        pipe = pipe_sd3.to(device)
+    # elif model_type_val == 'SD3':
+        # pipe = pipe_sd3.to(device)
     else:
         raise NotImplementedError(f"Model type {model_type_val} not implemented")