Hugging Face's logo

Spaces:
sujitpal
/
clip-rsicd-demo
Build error

App Files Community
8
sujitpal commited on
Commit
96ac3ab
·
1 Parent(s): 9cde513

fix: point to current model

Browse files
Files changed (2) hide show
  1. dashboard_image2image.py +1 -1
  2. dashboard_text2image.py +1 -1
dashboard_image2image.py CHANGED
@@ -10,7 +10,7 @@ from transformers import CLIPProcessor, FlaxCLIPModel
10
 
11
  BASELINE_MODEL = "openai/clip-vit-base-patch32"
12
  # MODEL_PATH = "/home/shared/models/clip-rsicd/bs128x8-lr5e-6-adam/ckpt-1"
13
- MODEL_PATH = "flax-community/clip-rsicd"
14
 
15
  # IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-baseline.tsv"
16
  # IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-bs128x8-lr5e-6-adam-ckpt-1.tsv"
 
10
 
11
  BASELINE_MODEL = "openai/clip-vit-base-patch32"
12
  # MODEL_PATH = "/home/shared/models/clip-rsicd/bs128x8-lr5e-6-adam/ckpt-1"
13
+ MODEL_PATH = "flax-community/clip-rsicd-v2"
14
 
15
  # IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-baseline.tsv"
16
  # IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-bs128x8-lr5e-6-adam-ckpt-1.tsv"
dashboard_text2image.py CHANGED
@@ -9,7 +9,7 @@ from transformers import CLIPProcessor, FlaxCLIPModel
9
 
10
  BASELINE_MODEL = "openai/clip-vit-base-patch32"
11
  # MODEL_PATH = "/home/shared/models/clip-rsicd/bs128x8-lr5e-6-adam/ckpt-1"
12
- MODEL_PATH = "flax-community/clip-rsicd"
13
 
14
  # IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-baseline.tsv"
15
  # IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-bs128x8-lr5e-6-adam-ckpt-1.tsv"
 
9
 
10
  BASELINE_MODEL = "openai/clip-vit-base-patch32"
11
  # MODEL_PATH = "/home/shared/models/clip-rsicd/bs128x8-lr5e-6-adam/ckpt-1"
12
+ MODEL_PATH = "flax-community/clip-rsicd-v2"
13
 
14
  # IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-baseline.tsv"
15
  # IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-bs128x8-lr5e-6-adam-ckpt-1.tsv"