Update app.py

app.py

@@ -14,6 +14,9 @@ from depth_anything_v2.dpt import DepthAnythingV2
 
 from huggingface_hub import hf_hub_download
 
+from huggingface_hub import login
+hf_token = os.environ.get("HF_TOKEN_GATED")
+login(token=hf_token)
 
 MAX_SEED = np.iinfo(np.int32).max
 
@@ -130,7 +133,7 @@ def resize_img(input_image, max_side=1024, min_side=768, size=None,
         input_image = Image.fromarray(res)
     return input_image
 
-@spaces.GPU()
+@spaces.GPU(duration=190)
 def infer(cond_in, image_in, prompt, inference_steps, guidance_scale, control_mode, control_strength, seed, progress=gr.Progress(track_tqdm=True)):
         
     control_mode_num = mode_mapping[control_mode]