Update app.py

app.py

@@ -49,7 +49,7 @@ high_threshold = 200
 # Models
 controlnet = ControlNetModel.from_pretrained("lllyasviel/sd-controlnet-canny", torch_dtype=torch.float16)
 pipe = DiffusionPipeline.from_pretrained(
-    "runwayml/stable-diffusion-v1-5", controlnet=controlnet, custom_pipeline="stable_diffusion_controlnet_img2img", safety_checker=None, torch_dtype=torch.float16
+    "mikesmodels/Waltz_with_Bashir_Diffusion", controlnet=controlnet, custom_pipeline="stable_diffusion_controlnet_img2img", safety_checker=None, torch_dtype=torch.float16
 )
 pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
 
@@ -125,7 +125,7 @@ def infer():
     #frames, _, _ = read_video(str("./spacex.mp4"), output_format="TCHW")
     #print(f"FRAME BEFORE stack: {frames[100]}")
     
-    prompt = "astronaut crew inside a spaceship, painting by Claude Monet"
+    prompt = "wltzwthbshr astronaut crew inside a spaceship"
     
     pil2diff_img = Image.open("./frame1.jpg")
     canny_image = get_canny_filter(pil2diff_img)