Add image resizing to display

app.py

@@ -46,6 +46,8 @@ def read_markdown(path, parent="./sections/"):
 
 def resize_height(image, new_height):
     h, w, c = image.shape
+    new_width = int(w * new_height / h)
+    return cv2.resize(image, (new_width, new_height))
 
 checkpoints = ["./ckpt/ckpt-60k-5999"]  # TODO: Maybe add more checkpoints?
 dummy_data = pd.read_csv("dummy_vqa_multilingual.tsv", sep="\t")
@@ -110,11 +112,11 @@ if uploaded_file is not None:
     state.image = np.array(Image.open(uploaded_file))
 
 
-state.image = 
+state.image = resize_height(state.image, 224)
 transformed_image = get_transformed_image(state.image)
 
 # Display Image
-col1.image(state.image, use_column_width="always")
+col1.image(state.image, use_column_width="auto")
 
 # Display Question
 question = col2.text_input(label="Question", value=state.question)