Bugfix: stop caching heatmap in localization

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()
@@ -139,7 +139,7 @@ def app():
 
     MAX_ITER = 1
 
-    col1, col2 = st.beta_columns([3, 1])
+    col1, col2 = st.columns([3, 1])
 
     with col2:
         pixel_size = st.selectbox("Pixel Size", options=range(10, 26, 5), index=2)
@@ -174,7 +174,10 @@ def app():
         gc.collect()
 
     elif image_url:
-        image = requests.get(image_url, stream=True,).raw
+        image = requests.get(
+            image_url,
+            stream=True,
+        ).raw
         image = Image.open(image).convert("RGB")
         with col1:
             st.image(image)