Update handler.py

handler.py

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