fix image loading

image2text.py

@@ -9,6 +9,11 @@ import requests
 import jax
 import gc
 
+headers = {
+    "User-Agent":
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.102 Safari/537.36 Edge/18.19582"
+}
+
 
 def app():
     st.title("From Image to Text")
@@ -63,7 +68,7 @@ def app():
                     text_embeds.extend(text_encoder(c, model, tokenizer)[0])
 
                 text_embeds = jnp.array(text_embeds)
-                response = requests.get(image_url)
+                response = requests.get(image_url, headers=headers, stream=True)
                 image = Image.open(BytesIO(response.content)).convert("RGB")
                 transform = get_image_transform(model.config.vision_config.image_size)
                 image_embed, _ = image_encoder(transform(image), model)
@@ -84,6 +89,6 @@ def app():
         gc.collect()
 
     elif image_url:
-        response = requests.get(image_url)
+        response = requests.get(image_url, headers=headers, stream=True)
         image = Image.open(BytesIO(response.content)).convert("RGB")
         st.image(image)

localization.py

@@ -13,6 +13,11 @@ import jax
 import gc
 
 
+headers = {
+    "User-Agent":
+    "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/70.0.3538.102 Safari/537.36 Edge/18.19582"
+}
+
 preprocess = transforms.Compose(
     [
         transforms.ToTensor(),
@@ -224,7 +229,8 @@ def app():
 
     elif image_url:
         image = requests.get(
-            image_url,
+            image_url, 
+            headers=headers,
             stream=True,
         ).raw
         image = Image.open(image).convert("RGB")