Upload app.py

app.py

@@ -61,7 +61,6 @@ print("SD Inpainting ready.")
 # COLORING- IMG2IMG Pipeline
 colour= StableDiffusionImg2ImgPipeline(
     "runwayml/stable-diffusion-v1-5",
-    torch_dtype=torch.float16,
     safety_checker=None,
     requires_safety_checker=False,
 )
@@ -220,7 +219,7 @@ def enhance_image(image, scale_factor):
 
 @spaces.GPU
 def colour_image(image, prompt, strength):
-    colour.to("cuda")
+    colour.to("cuda", torch.float16)
     if Image is None:
         raise gr.Error("Please Upload the Image")
     # Convert to RGB