fit result image to gallery

app.py

@@ -109,6 +109,7 @@ with gr.Blocks(css=custom_css) as interface:
         result_gallery = gr.Gallery(
             label="Results",
             rows=1,
+            object_fit="contain", 
             height=500,
         )
 
@@ -124,6 +125,7 @@ with gr.Blocks(css=custom_css) as interface:
         dff_gallery = gr.Gallery(
             label="Deep Feature Factorization",
             rows=1,
+            object_fit="contain", 
             height=800,
         )