Hugging Face's logo

Spaces:
AsmaAILab
/
empty-room-generation
Runtime error

App Files Community
AsmaAILab commited on
Commit
f1d6685
·
verified ·
1 Parent(s): 1a7479b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -64,8 +64,8 @@ model = Mask2FormerForUniversalSegmentation.from_pretrained("facebook/mask2forme
64
  model = model.to(DEVICE)
65
 
66
  # Load Depth model
67
- depth_image_processor = AutoImageProcessor.from_pretrained("LiheYoung/depth-anything-large-hf", torch_dtype=torch.float16)
68
- depth_model = AutoModelForDepthEstimation.from_pretrained("LiheYoung/depth-anything-large-hf", torch_dtype=torch.float16)
69
  depth_model = depth_model.to(DEVICE)
70
 
71
  def to_rgb(color: str) -> tuple:
 
64
  model = model.to(DEVICE)
65
 
66
  # Load Depth model
67
+ depth_image_processor = AutoImageProcessor.from_pretrained("LiheYoung/depth-anything-large-hf", torch_dtype=DTYPE)
68
+ depth_model = AutoModelForDepthEstimation.from_pretrained("LiheYoung/depth-anything-large-hf", torch_dtype=DTYPE)
69
  depth_model = depth_model.to(DEVICE)
70
 
71
  def to_rgb(color: str) -> tuple: