Add image URL option in MLM

apps/mlm.py

@@ -13,7 +13,8 @@ import os
 import matplotlib.pyplot as plt
 from mtranslate import translate
 from .utils import read_markdown
-
+import requests
+from PIL import Image
 from .model.flax_clip_vision_bert.modeling_clip_vision_bert import (
     FlaxCLIPVisionBertForMaskedLM,
 )
@@ -63,6 +64,7 @@ def app(state):
         with st.spinner("Loading model..."):
             mlm_state.mlm_model = load_model(mlm_checkpoints[0])
 
+
     if st.button(
         "Get a random example",
         help="Get a random example from the 100 `seeded` image-text pairs.",
@@ -79,6 +81,20 @@ def app(state):
         image = plt.imread(image_path)
         mlm_state.mlm_image = image
 
+    st.write("OR")
+    
+    query1 = st.text_input(
+        "Enter a URL to an image",
+        value="http://images.cocodataset.org/val2017/000000039769.jpg",
+    )
+    if st.button("Use this URL"):
+        image_data = requests.get(query1, stream=True).raw
+        image = np.asarray(Image.open(image_data))
+        print(image.shape)
+        mlm_state.mlm_image = image
+
+
+
     transformed_image = get_transformed_image(mlm_state.mlm_image)
 
     new_col1, new_col2 = st.beta_columns([5, 5])