Update app.py

app.py

@@ -14,7 +14,7 @@ model_id = "OpenVINO/LCM_Dreamshaper_v7-int8-ov"
 #pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False, safety_checker=safety_checker)
 pipeline = OVLatentConsistencyModelPipeline.from_pretrained(model_id, compile=False)
 
-batch_size, num_images, height, width = 1, 2, 512, 512
+batch_size, num_images, height, width = 1, 1, 512, 512
 pipeline.reshape(batch_size=batch_size, height=height, width=width, num_images_per_prompt=num_images)
 pipeline.load_textual_inversion("./badhandv4.pt", "badhandv4")
 pipeline.compile()
@@ -32,7 +32,7 @@ def infer(prompt, num_inference_steps):
         width = width,
         height = height,
         num_images_per_prompt=num_images,
-    ).images
+    ).images[0]
     
     return image