Update app.py

app.py

@@ -3,11 +3,12 @@ import numpy as np
 import streamlit as st
 import random
 import os
-from PIL import Image
+from PIL import Image, ImageOps
 from io import BytesIO
 from huggingface_hub import snapshot_download
 
 def random_crop(img, min_size=160, max_size=2048, ratio=5/8):
+    img = ImageOps.exif_transpose(img)
     width, height = img.size
 
     crop_width = random.randint(min_size, min(max_size, width))