Update app.py

app.py

@@ -3,9 +3,8 @@ from diffusers import StableDiffusionImg2ImgPipeline
 import torch
 import base64
 
-model = StableDiffusionImg2ImgPipeline.from_pretrained(
-    "black-forest-labs/FLUX.1-Kontext-dev", torch_dtype=torch.float16
-).to("cuda" if torch.cuda.is_available() else "cpu")
+model = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5")
+.to("cuda" if torch.cuda.is_available() else "cpu")
 
 def transform_image(image, prompt="place it"):
     result = model(prompt=prompt, image=image).images[0]