Add epoch=1-step=2184.ckpt weight file.

app.py

@@ -73,7 +73,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 feature_extractor = AutoFeatureExtractor.from_pretrained("hustvl/yolos-small", size=512, max_size=864)
 
 # Build model and load checkpoint
-checkpoint = 'fintune_traffic_object.ckpt'
+checkpoint = './checkpoints/epoch=1-step=2184.ckpt'
 model = Detr.load_from_checkpoint(checkpoint, lr=2.5e-5, weight_decay=1e-4)
 
 model.to(device)

checkpoints/epoch=1-step=2184.ckpt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:507d87450d7c714e2c360f4c92899c8140f9a54d5f193d2412a4494b9b95b4be
+size 368111967