Update app.py

app.py

@@ -313,9 +313,10 @@ def inference(input_prompt, input_category):
                         x_sample = torch.clamp((x_samples_ddim[0] + 1.0) / 2.0, min=0.0, max=1.0)
                         x_sample = 255. * rearrange(x_sample.cpu().numpy(), 'c h w -> h w c')
 
-                        Image.fromarray(x_sample.astype(np.uint8)).save("demo/demo.png")  
+                        # Image.fromarray(x_sample.astype(np.uint8)).save("demo/demo.png")
                         img = x_sample.astype(np.uint8)
                         generated_image = x_sample.astype(np.uint8)
+                        print(generated_image)
                         
                         class_name = trainclass