Update app.py

app.py

@@ -12,7 +12,7 @@ from huggingface_hub import snapshot_download
 from visualization import visualize_bbox
 
 # == download weights ==
-model_dir = snapshot_download('juliozhao/DocLayout-YOLO-DocStructBench', local_dir='./models/DocLayout-YOLO-DocStructBench')
+model_dir = snapshot_download('Tsiemonsma/app-archipelagos', local_dir='C:\yolov5\runs\train\exp5\weights')
 # == select device ==
 device = 'cuda' if torch.cuda.is_available() else 'cpu'