handle rotation, reduce input res to 720

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from PIL import Image,ImageDraw, ImageFont
+from PIL import Image,ImageDraw, ImageFont, ImageOps
 import sys
 import torch
 from util import Detection
@@ -26,10 +26,16 @@ def run_yolo(img):
 
     img_path = img
     img0 = Image.open(img_path).convert("RGB")
+
+    img0 = ImageOps.contain(img0, (720,720))
+    img0 = ImageOps.exif_transpose(img0)
+    
+    img0.save("img.jpg")
+
     draw = ImageDraw.Draw(img0)
 
     predictions = predict(age_model_ts, model, stride, names, pt, jit, onnx, engine, imgsz=[320, 320], conf_thres=0.5, iou_thres=0.45, save_conf=True,
-                    exist_ok=True, nosave=True, save_txt=False, source=img_path, project=None, name=None)
+                    exist_ok=True, nosave=True, save_txt=False, source="img.jpg", project=None, name=None)
 
     detections : list[Detection] = []
     for k, (bboxes, img)  in enumerate(predictions):
@@ -54,6 +60,7 @@ def run_yolo(img):
                 draw.rectangle(((det.xmin, det.ymin - 10), (det.xmax, det.ymin)), fill=(255,255,255))
                 draw.text((det.xmin, det.ymin - 10), det.class_name, fill=(0,0,0), font=roboto_font)
 
+    os.remove("img.jpg")
     return img0