Remove resizing

app.py

@@ -5,7 +5,7 @@ import json
 import os
 import numpy as np
 from streamlit.elements import markdown
-import cv2
+from PIL import Image
 from model.flax_clip_vision_bert.modeling_clip_vision_bert import (
     FlaxCLIPVisionBertForSequenceClassification,
 )
@@ -18,7 +18,6 @@ from utils import (
 )
 import matplotlib.pyplot as plt
 from mtranslate import translate
-from PIL import Image
 
 
 from session import _get_state
@@ -44,10 +43,11 @@ def read_markdown(path, parent="./sections/"):
     with open(os.path.join(parent, path)) as f:
         return f.read()
 
-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))
+
+# 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")
@@ -111,8 +111,6 @@ if uploaded_file is not None:
     state.image_file = os.path.join("images/val2014", uploaded_file.name)
     state.image = np.array(Image.open(uploaded_file))
 
-
-state.image = resize_height(state.image, 224)
 transformed_image = get_transformed_image(state.image)
 
 # Display Image

requirements.txt

@@ -4,5 +4,4 @@ git+https://github.com/huggingface/transformers.git
 torchvision==0.10.0
 mtranslate==1.8
 black==21.7b0
-flax==0.3.4 
-opencv-python==4.5.3.56
+flax==0.3.4