Update handler.py

handler.py

@@ -53,19 +53,19 @@ class EndpointHandler():
         
         prompt = data.pop("prompt", "")
 
-        if(method == "smooth"):
-            if encoded_image is not None:
-                image = self.decode_base64_image(encoded_image)
-                out = self.smooth_pipe(prompt, image=image).images[0]
-    
-                return out
-
         negative_prompt = data.pop("negative_prompt", "")
 
         method = data.pop("method", "slow")
         strength = data.pop("strength", 0.2)
         guidance_scale = data.pop("guidance_scale", 8.0)
         num_inference_steps = data.pop("num_inference_steps", 20)
+
+        if(method == "smooth"):
+            if encoded_image is not None:
+                image = self.decode_base64_image(encoded_image)
+                out = self.smooth_pipe(prompt, image=image).images[0]
+    
+                return out
         
         # process image
         if encoded_image is not None and encoded_mask_image is not None: