allow pixel size 10 and cache get_heatmap

localization.py

@@ -73,7 +73,7 @@ def gen_image_batch(image_url, image_size=224, pixel_size=10):
 
     return image_batch, masks
 
-
+@st.cache
 def get_heatmap(image_url, text, pixel_size=10, iterations=3):
     tokenizer = get_tokenizer()
     model = get_model()
@@ -142,7 +142,7 @@ def app():
     col1, col2 = st.beta_columns([3, 1])
 
     with col2:
-        pixel_size = st.selectbox("Pixel Size", options=range(15, 26, 5), index=1)
+        pixel_size = st.selectbox("Pixel Size", options=range(10, 26, 5), index=2)
 
         iterations = st.selectbox("Refinement Steps", options=range(3, 30, 3), index=0)