fixing

app.py

@@ -154,8 +154,8 @@ def combine_components_slice(model, gd, im1, im2, indices=None, sample_method='d
 def decompose_image_demo(im, model):
     sample_method = 'ddim'
     result = gen_image_and_components(MODELS[model], GD[sample_method], im, sample_method=sample_method, num_images=1, device=device)
-    result = Image.fromarray(result.permute(1, 2, 0).numpy())
-    return result
+    # result = Image.fromarray(result.permute(1, 2, 0).numpy())
+    return result.permute(1, 2, 0).numpy()
 
 
 def combine_images_demo(im1, im2, model):
@@ -164,6 +164,7 @@ def combine_images_demo(im1, im2, model):
     result = Image.fromarray(result.permute(1, 2, 0).numpy())
     if model == 'CelebA-HQ':
         return upscale_image(result, pipe)
+    return result
 
 
 def load_model(dataset, extra_kwargs={}, device='cuda'):
@@ -226,7 +227,7 @@ with gr.Blocks() as demo:
                 )
             
         with gr.Column():
-            decomp_output = gr.Image(type='pil')
+            decomp_output = gr.Image(type='numpy')
             decomp_button = gr.Button("Generate")
     
 
@@ -270,4 +271,4 @@ with gr.Blocks() as demo:
                        outputs=comb_output)
 
 
-demo.launch()
+demo.launch(debug=True)