Upload folder using huggingface_hub

app.py

@@ -11,7 +11,8 @@ import gradio as gr
 import random
 import torch
 import io
-from PIL import Image, ImageDraw, ImageFont
+import PIL.ImageOps
+from PIL import Image
 from transformers import AutoTokenizer, AutoModelForSequenceClassification
 
 load_dotenv()
@@ -174,7 +175,10 @@ def image_to_base64(file_path):
 
 def decode_base64_to_image(base64_str):
     image_data = base64.b64decode(base64_str)
-    return Image.open(io.BytesIO(image_data))
+    image = Image.open(io.BytesIO(image_data))
+    image = PIL.ImageOps.exif_transpose(image)
+    image = image.convert("RGB")
+    return image
 
 
 def generate_image(