Update app.py

app.py

@@ -68,9 +68,15 @@ def perform_neural_transfer(content_image_input, style_image_input, super_resolu
     stylized_image = tensor_to_image(stylized_image)
     content_image_input = tensor_to_image(content_image_input)
     stylized_image = stylized_image.resize(content_image_input.size)
+
+    print("super_resolution_type :")
+    print(super_resolution_type)
+    #print(super_resolution_type.value)
+    
     if super_resolution_type is "none":
         return stylized_image
     else:
+        print("call else :")
         stylized_image = inference(stylized_image, super_resolution_type)
         return stylized_image
 
@@ -80,7 +86,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         style_reference_input_gallery = gr.Gallery(list(style_urls.values()),
                         #width = 512,
-                        height = 768 + 32,
+                        height = 768 + 256,
                         label = "Style Image gallery (click to use)")
         with gr.Column():
             super_resolution_type = gr.Radio(["none" ,"base", "anime"], type="value", default="none", label="model used to super resolution the Image Transformed")
@@ -102,7 +108,7 @@ with gr.Blocks() as demo:
         image_click, style_reference_input_gallery, style_reference_input_image
     )
 
-    trans_button.click(perform_neural_transfer, [content_image_input, style_reference_input_image], trans_image_output)
+    trans_button.click(perform_neural_transfer, [content_image_input, style_reference_input_image, super_resolution_type], trans_image_output)
 
     gr.Examples(
         [