General clean up

geo_painting.py

@@ -1,15 +1,13 @@
 import torch
+from PIL import Image
 from diffusers import (
     StableDiffusionControlNetPipeline,
     UniPCMultistepScheduler,
     ControlNetModel
 )
-from diffusers.utils import load_image
-from PIL import Image
 
 
 class GeoPainting:
-
     DEFAULT_CONTROLNET_MODEL = "lllyasviel/control_v11f1p_sd15_depth"
     DEFAULT_DIFFUSER_MODEL = "geospatial_diffuser"
 
@@ -25,16 +23,11 @@ class GeoPainting:
         )
         self.pipe.scheduler = UniPCMultistepScheduler.from_config(self.pipe.scheduler.config)
         if torch.cuda.is_available():
-            print(":::::::CUDA AVAILABLE::GPU:HUGGINGFACE:::::::")
             self.pipe.enable_model_cpu_offload()
             self.pipe.enable_xformers_memory_efficient_attention()
 
     def generate_painting(self, input_promp, control_image):
-        print(":::::::::::::::::")
-        print(type(control_image))
         image = Image.fromarray(control_image.astype('uint8'))
-        print(type(image))
-
         output = self.pipe(
             input_promp,
             image,