Update app.py

app.py

@@ -3,9 +3,20 @@ from transformers import pipeline
 import torch
 from diffusers import StableDiffusionPipeline
 
-def get_completion(prompt,params):
-    return pipeline(prompt=prompt, height=params['height'], width=params['width'], num_inference_steps=params['num_inference_steps'], guidance_scale=params['guidance_scale'], negative_prompt=params['negative_prompt'])['sample'][0]
-    # return pipeline(prompt, params)['sample'][0]
+def get_completion(inputs, parameters=None):
+    data = { "inputs": inputs }
+    if parameters is not None:
+        data.update({"parameters": parameters})
+    return pipeline(prompt, params)['sample'][0]
+    # response = requests.request("POST",
+    #                             ENDPOINT_URL,
+    #                             headers=headers,
+    #                             data=json.dumps(data))
+    # return json.loads(response.content.decode("utf-8"))
+    
+# def get_completion(prompt,params):
+#     return pipeline(prompt=prompt, height=params['height'], width=params['width'], num_inference_steps=params['num_inference_steps'], guidance_scale=params['guidance_scale'], negative_prompt=params['negative_prompt'])['sample'][0]
+#     # return pipeline(prompt, params)['sample'][0]
 
 def generate(prompt,negative_prompt,steps,guidance,width,height):
     params = {