TheFrenchDemos commited on
Commit
e8c243a
·
verified ·
1 Parent(s): bd0320b

Update app.py

Browse files

changement de la fonction load_model à cause du runtime error

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ model = None
15
 
16
  def load_model():
17
  """Load DINOv2 model"""
18
- torch.hub.set_dir('static') # Cache local des modèles
19
  model = torch.hub.load('facebookresearch/dinov2', 'dinov2_vits14')
20
  model.eval()
21
  model.to(torch.device('cuda' if torch.cuda.is_available() else 'cpu'))
 
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()
21
  model.to(torch.device('cuda' if torch.cuda.is_available() else 'cpu'))