Update app.py

app.py

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