Update app.py

app.py

@@ -100,12 +100,12 @@ def king(type ,
             print(f"AFTER: {instruction} ")
         guidance_scale2=(guidance_scale/2)
         if fast:
-            image = pipe_fast(prompt = instruction,
+            refine = pipe_fast(prompt = instruction,
             guidance_scale = guidance_scale2, 
             num_inference_steps = int(steps/2.5),
             width = width, height = height,
-            generator = generator, output_type="latent",
-            ).images
+            generator = generator,
+            ).images[0]
         else:            
             image = pipe_fast( prompt = instruction,
             negative_prompt=negative_prompt,
@@ -115,12 +115,12 @@ def king(type ,
             generator = generator, output_type="latent",
             ).images 
 
-        refine = refiner( prompt=instruction,
-                negative_prompt = negative_prompt,
-                guidance_scale = guidance_scale,
-                num_inference_steps=  steps,
-                image=image, generator=generator,
-        ).images[0]        
+            refine = refiner( prompt=instruction,
+                    negative_prompt = negative_prompt,
+                    guidance_scale = guidance_scale,
+                    num_inference_steps=  steps,
+                    image=image, generator=generator,
+            ).images[0]        
         return seed, refine
 
 client = InferenceClient()