Spaces:
Runtime error
Runtime error
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'))
|