Update app.py

app.py

@@ -4,8 +4,8 @@ import torch
 from diffusers import DiffusionPipeline
     
 def get_completion(prompt,params):
-    return pipeline(prompt=prompt, height=params['height'], width=params['width'], num_inference_steps=int(params['num_inference_steps']), guidance_scale=params['guidance_scale'])['sample'][0]
-
+    # return pipeline(prompt=prompt, height=params['height'], width=params['width'], num_inference_steps=int(params['num_inference_steps']), guidance_scale=params['guidance_scale'])['sample'][0]
+    return pipeline(prompt=prompt, height=params['height'], width=params['width'], num_inference_steps=int(params['num_inference_steps']), guidance_scale=params['guidance_scale']).images[0]
 def generate(prompt,steps,guidance,width,height):
     params = {
         "num_inference_steps": steps,