fix: removing commented out code since runs on HF spaces

dashboard_featurefinder.py

@@ -15,14 +15,10 @@ from PIL import Image
 import utils
 
 BASELINE_MODEL = "openai/clip-vit-base-patch32"
-# MODEL_PATH = "/home/shared/models/clip-rsicd/bs128x8-lr5e-6-adam/ckpt-1"
 MODEL_PATH = "flax-community/clip-rsicd-v2"
 
-# IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-baseline.tsv"
-# IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-bs128x8-lr5e-6-adam-ckpt-1.tsv"
 IMAGE_VECTOR_FILE = "./vectors/test-bs128x8-lr5e-6-adam-ckpt-1.tsv"
 
-# IMAGES_DIR = "/home/shared/data/rsicd_images"
 IMAGES_DIR = "./images"
 DEMO_IMAGES_DIR = "./demo-images"
 
@@ -113,7 +109,6 @@ def app():
         to look for, this will show the original image, and you will get more ideas
         for features that you can ask the model to identify.
     """)
-    # buf = st.file_uploader("Upload Image for Analysis", type=["png", "jpg"])
     image_file = st.selectbox(
         "Sample Image File",
         options=[
@@ -132,8 +127,6 @@ def app():
     searched_feature = st.text_input("Feature to find", value="beach")
 
     if st.button("Find"):
-        # print("image_file:", image_file)
-        # print("image_url:", image_url)
         if image_file.startswith("--"):
             image = download_and_prepare_image(image_url)
         else:
@@ -144,9 +137,7 @@ def app():
         else:
             st.image(image, caption="Input Image")
             st.markdown("---")
-            # print("image.shape:", image.shape)
             num_rows, num_cols, patches = split_image(image)
-            # print("num_rows, num_cols, num(patches:", num_rows, num_cols, len(patches), patches[0].shape)
             image_preprocessor = Compose([
                 ToPILImage(),
                 Resize(224)
@@ -167,9 +158,3 @@ def app():
                         patch_ranks[pid] + 1, searched_feature, patch_probs[pid])
                     col.image(patches[pid], caption=caption)
                     pid += 1
-                # row_patches = patches[i * num_cols : (i + 1) * num_cols]
-                # row_probs = patch_probs[i * num_cols : (i + 1) * num_cols]
-                # row_ranks = patch_ranks[i * num_cols : (i + 1) * num_cols]
-                # captions = ["p({:s})={:.3f}, rank={:d}".format(searched_feature, p, r + 1) 
-                #     for p, r in zip(row_probs, row_ranks)]
-                # st.image(row_patches, caption=captions)

dashboard_image2image.py

@@ -11,14 +11,10 @@ from transformers import CLIPProcessor, FlaxCLIPModel
 import utils
 
 BASELINE_MODEL = "openai/clip-vit-base-patch32"
-# MODEL_PATH = "/home/shared/models/clip-rsicd/bs128x8-lr5e-6-adam/ckpt-1"
 MODEL_PATH = "flax-community/clip-rsicd-v2"
 
-# IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-baseline.tsv"
-# IMAGE_VECTOR_FILE = "/home/shared/data/vectors/test-bs128x8-lr5e-6-adam-ckpt-1.tsv"
 IMAGE_VECTOR_FILE = "./vectors/test-bs128x8-lr5e-6-adam-ckpt-1.tsv"
 
-# IMAGES_DIR = "/home/shared/data/rsicd_images"
 IMAGES_DIR = "./images"
 
 
@@ -50,10 +46,7 @@ def download_and_prepare_image(image_url):
         image_raw = requests.get(image_url, stream=True,).raw
         image = Image.open(image_raw).convert("RGB")
         width, height = image.size
-        # print("width, height:", width, height)
         resize_mult = width / 224 if width < height else height / 224
-        # print("resize_mult:", resize_mult)
-        # print("resize:", width // resize_mult, height // resize_mult)
         image = image.resize((int(width // resize_mult), 
                               int(height // resize_mult)))
         width, height = image.size
@@ -61,11 +54,9 @@ def download_and_prepare_image(image_url):
         top = int((height - 224) // 2)
         right = int((width + 224) // 2)
         bottom = int((height + 224) // 2)
-        # print("LTRB:", left, top, right, bottom)
         image = image.crop((left, top, right, bottom))
         return image
     except Exception as e:
-        # print(e)
         return None
 
 def app():
@@ -137,7 +128,7 @@ def app():
 
     image_url = st.text_input(
         "OR provide an image URL",
-        value="https://media.wired.com/photos/5a8c80647b7bd44d86b88077/master/w_2240,c_limit/Satellite-FINAL.jpg")
+        value="https://static.eos.com/wp-content/uploads/2019/04/Main.jpg")
     
     submit_button = st.button("Find Similar")