add additional parameters

README.md

@@ -7,6 +7,12 @@ widget:
   - src: >-
       https://huggingface.co/datasets/mishig/sample_images/resolve/main/canny-edge.jpg
     prompt: Girl with Pearl Earring
+    num_inference_steps: ''
+    guidance_scale: ''
+    negative_prompt: ''
+    height: ''
+    width: ''
+    manual_seed: ''
 duplicated_from: stabilityai/stable-diffusion-2-1-base
 ---
 

handler.py

@@ -51,7 +51,7 @@ class EndpointHandler():
 
         out = None
 
-        if data.get("url"):
+        if data.get("src"):
             generator = torch.Generator(device='cuda')
             generator.manual_seed(manual_seed)
             # run img2img pipeline