Update app.py

app.py

@@ -79,21 +79,31 @@ def king(type ,
         text_cfg_scale = text_cfg_scale
         image_cfg_scale = image_cfg_scale
         input_image = input_image
-
         steps=steps
         generator = torch.manual_seed(seed)
         output_image = pipe_edit(
             instruction, image=input_image,
             guidance_scale=text_cfg_scale, image_guidance_scale=image_cfg_scale,
-            num_inference_steps=steps, generator=generator).images[0]
-        return seed, output_image
+            num_inference_steps=steps, generator=generator, output_type="latent",
+        ).images         
+
+        refine = refiner(
+            prompt=instruction,
+            guidance_scale=6,
+            num_inference_steps=25,
+            image=image,
+            width = width, 
+            height = height,
+            generator=generator,
+        ).images[0]  
+        return seed, refine
     else :
         if randomize_seed:
             seed = random.randint(0, 99999)
         generator = torch.Generator().manual_seed(seed)
         image = pipe(
             prompt = instruction,
-            guidance_scale =6, 
+            guidance_scale =7, 
             num_inference_steps = steps, 
             width = width, 
             height = height,