Update app.py

app.py

@@ -4,14 +4,23 @@ from PIL import Image
 import matplotlib.pyplot as plt
 from safetensors.torch import load_model
 from transformers import pipeline
-import torch
+import torch, cv2
 from torch import nn
+import numpy as np
 from torch.nn import functional as func_nn
 from einops import rearrange
 from huggingface_hub import PyTorchModelHubMixin
 from torchvision import models
 
 
+def read_image(img, img_size=config.image_size):
+    img = np.array(img)
+    img = cv2.resize(img, (img_size, img_size)) # resize to mathc model input here
+    img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
+    img = img / 255.0
+
+    return img
+    
 # main model network
 class SiameseNetwork(nn.Module, PyTorchModelHubMixin):
     def __init__(self):