Fix height and width resize

app.py

@@ -45,7 +45,7 @@ model.eval()
 
 
 # transforms
-downsize_t = transforms.Resize(config["downsize_res"], antialias=True)
+downsize_t = transforms.Resize((config["downsize_res"], config["downsize_res"]), antialias=True)
 transform = transforms.Compose(
     [
         transforms.ToTensor(),