Spaces:
Sleeping
Sleeping
Update app.py
Browse fileschangement de la fonction load_model à cause du runtime error
app.py
CHANGED
@@ -15,7 +15,7 @@ model = None
|
|
15 |
|
16 |
def load_model():
|
17 |
"""Load DINOv2 model"""
|
18 |
-
torch.hub.set_dir(
|
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'))
|