jichao commited on
Commit
48e1f93
·
1 Parent(s): 353aff6

add vit-s-dino-1010-e50

Browse files
app.py CHANGED
@@ -22,12 +22,12 @@ MODEL_CONFIGS = {
22
  "description": "ViT-Base/16 (Grayscale Input)"
23
  },
24
  # --- Add more model configurations here ---
25
- # "another_model_name": {
26
- # "path": "models/another_checkpoint.pth",
27
- # "timm_id": "vit_small_patch16_224",
28
- # "in_chans": 3, # Example: RGB model
29
- # "description": "ViT-Small/16 (RGB Input)"
30
- # },
31
  }
32
 
33
  # Global dictionary to store loaded models
 
22
  "description": "ViT-Base/16 (Grayscale Input)"
23
  },
24
  # --- Add more model configurations here ---
25
+ "mars-ctx-vits-dino-1010-50": {
26
+ "path": "models/vit-s-dino-v1-1010-e50-use-this.pth",
27
+ "timm_id": "vit_small_patch16_224",
28
+ "in_chans": 1,
29
+ "description": "ViT-Small/16 (Grayscale Input)"
30
+ },
31
  }
32
 
33
  # Global dictionary to store loaded models
models/vit-s-dino-v1-1010-e50-use-this.pth ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:d1ed930ba752af0aa86a33ef4feace75ea5bb1420dbce96dbb95d8a8f04c7994
3
+ size 85934443
requirements.txt CHANGED
@@ -2,3 +2,4 @@ torch
2
  timm
3
  torchvision
4
  Pillow
 
 
2
  timm
3
  torchvision
4
  Pillow
5
+ gradio>5.0.0