Update app.py

app.py

@@ -70,6 +70,17 @@ title = "6D Rotation Representation for Unconstrained Head Pose Estimation"
 description = "Gradio demo for 6DRepNet. To use it, simply click the camera picture. Read more at the links below."
 article = "<div style='text-align: center;'><a href='https://github.com/thohemp/6DRepNet' target='_blank'>Github Repo</a> | <a href='https://arxiv.org/abs/2202.12555' target='_blank'>Paper</a></div>"
 
+image_flip_css = """
+.input-image .image-preview img {
+    -webkit-transform: scaleX(-1);
+    transform: scaleX(-1) !important;
+}
+
+.output-image img {
+    -webkit-transform: scaleX(-1);
+    transform: scaleX(-1) !important;
+}
+"""
   
 iface = gr.Interface(
     fn=predict, 
@@ -79,8 +90,7 @@ iface = gr.Interface(
     title=title,
     description=description,
     article=article,
-    css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
-
+    css = image_flip_css
 )
 
 iface.launch()