prthm11 commited on
Commit
6e18f95
·
verified ·
1 Parent(s): b2229e8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -325,7 +325,7 @@ def init_dinov2(model_name: str = DINOV2_MODEL, device: torch.device = DEVICE):
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
 
 
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, use_fast=True)
329
  _dinov2_model = AutoModel.from_pretrained(model_name)
330
  _dinov2_model.eval().to(device)
331