default parameter

README.md

@@ -16,7 +16,7 @@ inference: true
 
 ```json
 {
-    "prompt": "A prompt used for image generation",
+    "inputs": "A prompt used for image generation",
     "negative_prompt": "low res, bad anatomy, worst quality, low quality",
     "image_depth_map" : "iVBORw0KGgoAAAANSUhEUgAAAgAAAAIACAIAAAB7GkOtAAAABGdBTUEAALGPC",
     "steps": 25,
@@ -32,4 +32,4 @@ SD_MODEL no tested
 
 default => Lykon/dreamshaper-8
 
-dreamshaper || juggernaut || realistic-vision || rev-animated
+default || dreamshaper || juggernaut || realistic-vision || rev-animated

handler.py

@@ -50,7 +50,7 @@ class EndpointHandler():
 
         # hyperparamters
         sd_model = data.pop("sd_model", None)
-        prompt = data.pop("prompt", None)
+        prompt = data.pop("inputs", None)
         negative_prompt = data.pop("negative_prompt", None)
         image_depth_map = data.pop("image_depth_map", None)
         steps = data.pop("steps", 25)