Add image URL option to VQA

apps/mlm.py

@@ -90,7 +90,6 @@ def app(state):
     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
 
 

apps/vqa.py

@@ -10,6 +10,8 @@ import streamlit as st
 import numpy as np
 import pandas as pd
 import os
+import requests
+from PIL import Image
 import matplotlib.pyplot as plt
 import json
 
@@ -91,6 +93,17 @@ def app(state):
         image = plt.imread(image_path)
         vqa_state.vqa_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))
+        vqa_state.mlm_image = image
+
     transformed_image = get_transformed_image(vqa_state.vqa_image)
 
     new_col1, new_col2 = st.beta_columns([5, 5])