Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ app = Flask(__name__)
|
|
14 |
model = None
|
15 |
|
16 |
def load_model():
|
17 |
-
"""Load DINOv2
|
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()
|