Update app.py

app.py

@@ -172,6 +172,25 @@ class ImageProcessor:
 
         return pipe
 
+    def process_image(self, input_image, prompt, negative_prompt, resolution=2048, num_inference_steps=50, guidance_scale=3, strength=0.35, hdr=0):
+        condition_image = resize_and_upscale(input_image, resolution)
+        condition_image = create_hdr_effect(condition_image, hdr)
+
+        result = self.pipe(
+            prompt=prompt,
+            negative_prompt=negative_prompt,
+            image=condition_image,
+            control_image=condition_image,
+            width=condition_image.size[0],
+            height=condition_image.size[1],
+            strength=strength,
+            num_inference_steps=num_inference_steps,
+            guidance_scale=guidance_scale,
+            generator=torch.manual_seed(0),
+        ).images[0]
+
+        return result
+
 
 image_processor = ImageProcessor()