prthm11 commited on
Commit
b2229e8
·
verified ·
1 Parent(s): 9fa9940

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -324,7 +324,8 @@ def init_dinov2(model_name: str = DINOV2_MODEL, device: torch.device = DEVICE):
324
  """Lazy-initialize DINOv2 processor & model (call once before embedding)."""
325
  global _dinov2_processor, _dinov2_model
326
  if _dinov2_processor is None or _dinov2_model is None:
327
- _dinov2_processor = AutoImageProcessor.from_pretrained(model_name)
 
328
  _dinov2_model = AutoModel.from_pretrained(model_name)
329
  _dinov2_model.eval().to(device)
330
 
 
324
  """Lazy-initialize DINOv2 processor & model (call once before embedding)."""
325
  global _dinov2_processor, _dinov2_model
326
  if _dinov2_processor is None or _dinov2_model is None:
327
+ # _dinov2_processor = AutoImageProcessor.from_pretrained(model_name)
328
+ _dinov2_processor = AutoImageProcessor.from_pretrained(model_name_or_path, use_fast=True)
329
  _dinov2_model = AutoModel.from_pretrained(model_name)
330
  _dinov2_model.eval().to(device)
331