TheFrenchDemos commited on
Commit
51016cb
·
verified ·
1 Parent(s): d91cf7d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ app = Flask(__name__)
14
  model = None
15
 
16
  def load_model():
17
- """Load DINOv2 model"""
18
  torch.hub.set_dir("/tmp/torch_cache") # Dossier temporaire autorisé
19
  model = torch.hub.load('facebookresearch/dinov2', 'dinov2_vits14')
20
  model.eval()
 
14
  model = None
15
 
16
  def load_model():
17
+ """Load DINOv2 model"""
18
  torch.hub.set_dir("/tmp/torch_cache") # Dossier temporaire autorisé
19
  model = torch.hub.load('facebookresearch/dinov2', 'dinov2_vits14')
20
  model.eval()