Update app.py

app.py

@@ -8,7 +8,6 @@ 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))
@@ -52,6 +51,7 @@ file_name = st.file_uploader("Choose an image...")
 if file_name is not None:
     col1, col2 = st.columns(2)
     image = Image.open(file_name)
+    image = ImageOps.exif_transpose(image)
     
     if image.size != (200, 200) or image.mode != 'RGB':
         #image = random_crop(image)