Update app.py

app.py

@@ -15,7 +15,6 @@ from diffusers.utils import load_image
 from pipeline_flux_control_removal import FluxControlRemovalPipeline
 
 torch.set_grad_enabled(False)
-os.environ['GRADIO_TEMP_DIR'] = './tmp'
 device = "cuda"
 print(device)
 image_path = mask_path = None
@@ -52,7 +51,7 @@ image_examples = [
     ]
     
 ]
-@spaces.GPU
+@spaces.GPU(duration=120)
 def load_model(base_model_path, lora_path):
     global pipe
     transformer = FluxTransformer2DModel.from_pretrained(base_model_path, subfolder='transformer', torch_dtype=torch.bfloat16)
@@ -87,7 +86,7 @@ def load_model(base_model_path, lora_path):
     pipe.load_lora_weights(lora_path, weight_name="pytorch_lora_weights.safetensors")
     gr.Info(str(f"Model loading: {int((100 / 100) * 100)}%"))
     
-@spaces.GPU
+@spaces.GPU(duration=120)
 def set_seed(seed):
     torch.manual_seed(seed)
     torch.cuda.manual_seed(seed)
@@ -95,7 +94,7 @@ def set_seed(seed):
     np.random.seed(seed)
     random.seed(seed)
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def predict(
     input_image,
     prompt,