aniket47 commited on
Commit
3c28cb2
·
1 Parent(s): 52c65ee

Final update to use Intel DPT BEIT Large 512 model as requested

Browse files
Files changed (1) hide show
  1. models/depth_processor.py +2 -2
models/depth_processor.py CHANGED
@@ -28,8 +28,8 @@ class DepthProcessor:
28
  try:
29
  logger.info(f"🔄 Loading DPT model on {self.device}...")
30
 
31
- # Load processor and model (using smaller model for HF Spaces)
32
- model_name = "Intel/dpt-swinv2-tiny-256" # Smaller, faster model
33
  self.processor = DPTImageProcessor.from_pretrained(model_name)
34
  self.model = DPTForDepthEstimation.from_pretrained(model_name)
35
  self.model.to(self.device)
 
28
  try:
29
  logger.info(f"🔄 Loading DPT model on {self.device}...")
30
 
31
+ # Load processor and model
32
+ model_name = "Intel/dpt-beit-large-512" # Using the large model as requested
33
  self.processor = DPTImageProcessor.from_pretrained(model_name)
34
  self.model = DPTForDepthEstimation.from_pretrained(model_name)
35
  self.model.to(self.device)