Add media screen

app.py

@@ -84,7 +84,8 @@ description = "Demo for Microsoft's LayoutLMv2, a Transformer for state-of-the-a
 article = "<p style='text-align: center'><a href='https://arxiv.org/abs/2012.14740' target='_blank'>LayoutLMv2: Multi-modal Pre-training for Visually-Rich Document Understanding</a> | <a href='https://github.com/microsoft/unilm' target='_blank'>Github Repo</a></p>"
 examples =[['document.png']]
 
-css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
+css = "@media screen and (max-width: 600px) { .output_image, .input_image {height:20rem !important; width: 100% !important;} }"
+#css = ".output_image, .input_image {height: 40rem !important; width: 100% !important;}"
 # css = ".output_image, .input_image {height: 600px !important}"
 
 iface = gr.Interface(fn=process_image,