Upload app.py

app.py

@@ -60,15 +60,12 @@ print("SD Inpainting ready.")
 
 # COLORING- IMG2IMG Pipeline
 colour= StableDiffusionImg2ImgPipeline(
-    vae= inpaint.vae,
-    text_encoder= inpaint.text_encoder,
-    tokenizer= inpaint.tokenizer,
-    unet= inpaint.unet,
-    scheduler= inpaint.scheduler,
-    safety_checker= None,
-    feature_extractor= None,
-    requires_safety_checker= False,
-).to("cuda")
+    "runwayml/stable-diffusion-v1-5",
+    torch_dtype=torch.float16,
+    safety_checker=None,
+    requires_safety_checker=False,
+)
+print("colour model ready!")
 
 
 #  BLIP 
@@ -223,6 +220,7 @@ def enhance_image(image, scale_factor):
 
 @spaces.GPU
 def colour_image(image, prompt, strength):
+    colour.to("cuda")
     if Image is None:
         raise gr.Error("Please Upload the Image")
     # Convert to RGB