Update engine.py

engine.py

@@ -31,6 +31,10 @@ class Prodia:
         response = self._post(f"{self.base}/sdxl/generate", params)
         return response.json()
 
+    def upscale_image(self, params):
+        response = self._post(f"{self.base}/upscale", params)
+        return response.json()
+
     def get_job(self, job_id):
         response = self._get(f"{self.base}/job/{job_id}")
         return response.json()
@@ -158,7 +162,6 @@ def transform_sd(image, model, prompt, denoising_strength, negative_prompt, step
 
 
 def controlnet_sd(image, controlnet_model, controlnet_module, threshold_a, threshold_b, resize_mode, prompt, negative_prompt, steps, cfg_scale, seed, sampler, width, height):
-    print(image)
     image_url = prodia_client.upload(image)
     result = prodia_client.sd_transform({
         "imageUrl": image_url,
@@ -181,5 +184,16 @@ def controlnet_sd(image, controlnet_model, controlnet_module, threshold_a, thres
 
     return job["imageUrl"]
 
+def upscale(image, scale_by):
+    image_url = prodia_client.upload(image)
+    result = prodia_client.upscale_image({
+        'imageUrl': image_url,
+        'resize': scale_by
+    })
+
+    job = prodia_client.wait(result)
+
+    return job["imageUrl"]
+
 def get_models():
     return prodia_client.list_models()