Update handler.py

handler.py

@@ -19,11 +19,10 @@ class EndpointHandler():
   
   def __call__(self, data:Any) -> List[List[Dict[str, float]]]:
     print(data)
+    inputs = data.pop("inputs",data)
     print(device)
     with autocast(device.type):
-        print("feeding prompt into model");
-      image = self.pipe(prompt=data, guidance_scale=7.5).images[0]
-      print("model response" generated);
+      image = self.pipe(inputs, guidance_scale=7.5).images[0]
       
     buffered = BytesIO()
     image.save(buffered, format="JPEG")