Update app.py

app.py

@@ -43,7 +43,8 @@ model = DepthAnythingV2(**model_configs[encoder])
 filepath = hf_hub_download(repo_id=f"depth-anything/Depth-Anything-V2-Large", filename=f"depth_anything_v2_vitl.pth", repo_type="model")
 state_dict = torch.load(filepath, map_location="cpu")
 model.load_state_dict(state_dict)
-model = model.to(DEVICE).eval()
+# model = model.to(DEVICE).eval()
+model = model.eval()
 
 import torch
 from diffusers.utils import load_image
@@ -180,6 +181,8 @@ def infer(cond_in, image_in, prompt, inference_steps, guidance_scale, control_mo
         guidance_scale=guidance_scale,
         generator=torch.manual_seed(seed),
     ).images[0]
+
+    torch.cuda.empty_cache() 
     
     return image, control_image, gr.update(visible=True)