Update app.py

app.py

@@ -37,7 +37,17 @@ from urllib.request import urlretrieve
 
 from scipy.interpolate import LinearNDInterpolator
 from imageio import imread, imwrite
+from diffusers import StableDiffusionImg2ImgPipeline
 
+def diffuse(source_img, prompt):
+    model_id = "runwayml/stable-diffusion-v1-5"
+    pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+    pipe = pipe.to("cuda")
+    
+    prompt = "space crew in a spaceship by Claude Monet"
+    image = pipe(prompt, source_img).images[0]  
+    print(f"DIFFUSED IMG: {image}")
+    image.save("astronaut_rides_horse.png")
 
 def write_flo(flow, filename):
     """