exif, more examples

app.py

@@ -2,9 +2,13 @@ import gradio as gr
 import sys, os
 import torch
 import matplotlib.pylab as plt
+from PIL import ImageOps
 
 def pump_matching(img1, img2, trained_with_st=False, scale=300, max_scale=1, max_rot=0, use_gpu=False):
 
+    img1 = ImageOps.exif_transpose(img1)
+    img2 = ImageOps.exif_transpose(img2)
+
     use_singlescale = max_scale==1 and max_rot==0
     if use_singlescale: # single 
         from test_singlescale import Main, arg_parser
@@ -73,6 +77,8 @@ iface = gr.Interface(
     description=description,
     article=article,
     examples=[
+        ['datasets/gradio_demo/cat_src.jpg','datasets/gradio_demo/cat_tgt.jpg',False],#,400,1,0]+([True] if has_cuda else []),
+        ['datasets/gradio_demo/food_src.jpg','datasets/gradio_demo/food_tgt.jpg',False],#,400,1,0]+([True] if has_cuda else []),
         ['datasets/demo_warp/mountains_src.jpg','datasets/demo_warp/mountains_tgt.jpg',False],#,400,1,0]+([True] if has_cuda else []),
     ]
 )