Update app.py

app.py

@@ -15,7 +15,9 @@ device = "cuda"
 sdxl = InferenceClient(model="stabilityai/stable-diffusion-xl-base-1.0", token=os.environ['HF_TOKEN'])
 print('sdxl loaded')
 
-pipeline2Image = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtypes=torch.bfloat16).to(device)
+"kandinsky-community/kandinsky-2-2-decoder"
+#pipeline2Image = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-1.0", torch_dtypes=torch.bfloat16).to(device)
+pipeline2Image = DiffusionPipeline.from_pretrained("kandinsky-community/kandinsky-2-2-decoder", torch_dtypes=torch.bfloat16).to(device)
 print("pipeline 2 image loaded")
 
 MAX_SEED = np.iinfo(np.int32).max