jichao commited on
Commit
8dbe4eb
·
1 Parent(s): 87188fa
app.py CHANGED
@@ -10,7 +10,7 @@ import os
10
  DEFAULT_MODEL_NAME = "mars-ctx-vitb-0217"
11
  MODEL_CONFIGS = {
12
  "mars-ctx-vitb-0217": {
13
- "path": "models/checkpoint-300.pth",
14
  "timm_id": "vit_base_patch16_224",
15
  "in_chans": 1,
16
  "description": "ViT-Base/16 (Grayscale Input)"
 
10
  DEFAULT_MODEL_NAME = "mars-ctx-vitb-0217"
11
  MODEL_CONFIGS = {
12
  "mars-ctx-vitb-0217": {
13
+ "path": "models/0217-checkpoint-300.pth",
14
  "timm_id": "vit_base_patch16_224",
15
  "in_chans": 1,
16
  "description": "ViT-Base/16 (Grayscale Input)"
models/{checkpint-300.pth → 0217-checkpoint-300.pth} RENAMED
File without changes